def test_catch_error():
"""
If code raises an error, we should catch it!
Raised exceptions also halt execution of all future cells.
"""
with open(os.path.join(here, 'notebooks/partial-error.ipynb')) as f:
nb = json.load(f)
with pytest.raises(NameError):
execute_notebook(nb, ignore_errors=False)
def test_line_continuation():
"""
Test line continuations work in notebooks
"""
with open(os.path.join(here, 'notebooks/line-continuations.ipynb')) as f:
nb = json.load(f)
return_env = execute_notebook(nb)
assert return_env['return_2']() == 2
def test_okgrade_magic():
"""
Test we can execute notebooks with injected global variables
"""
with open(os.path.join(here, 'notebooks/injected-magic.ipynb')) as f:
nb = json.load(f)
return_env = execute_notebook(nb, initial_env={'__GOFER_GRADER__': True})
assert return_env['gofer_grader_present'] == True
def test_simple_execute():
"""
Test we can execute notebooks & see the resulting environment
"""
with open(os.path.join(here, 'notebooks/simplest-notebook.ipynb')) as f:
nb = json.load(f)
return_env = execute_notebook(nb)
assert return_env['the_number_1'] == 1
assert return_env['return_1']() == 1
def test_ignore_error():
"""
Ignoring errors should work similar to notebooks
The cell with the error should be executed until the error is
raised, and then the next cell should be executed.
"""
with open(os.path.join(here, 'notebooks/partial-error.ipynb')) as f:
nb = json.load(f)
return_env = execute_notebook(nb, ignore_errors=True)
assert return_env['a'] == 5
assert return_env['b'] == 6
assert return_env['c'] == 7
assert 'd' not in return_env
assert 'e' not in return_env
assert return_env['f'] == 9