Check strings in jupyter notebooks

This commit is contained in:
cyyc1 2022-10-05 02:42:08 -07:00
parent 6b03546fc3
commit cf898e2ce2
7 changed files with 528 additions and 42 deletions

View file

@ -0,0 +1,16 @@
from pre_commit_hooks.string_fixer_for_jupyter_notebooks import main
from testing.util import get_resource_path
def test_rewrite(tmpdir):
with open(get_resource_path('before.ipynb')) as f:
before_contents = f.read()
with open(get_resource_path('after.ipynb')) as f:
after_contents = f.read()
path = tmpdir.join('file.ipynb')
path.write(before_contents)
retval = main([str(path)])
assert path.read() == after_contents
assert retval == 1