test_demo_reweighing_preproc.py

# Adapted from:
# http://www.christianmoscardi.com/blog/2016/01/20/jupyter-testing.html

from notebook_runner import notebook_run
import os


def test_reweighing_preproc():
    nb, errors = notebook_run(os.path.join(
            os.path.dirname(os.path.abspath(__file__)),
            '..', 'examples', 'demo_reweighing_preproc.ipynb'))

    if len(errors) > 0:
        for err in errors:
            for tbi in err['traceback']:
                print(tbi)
        raise AssertionError("errors in notebook testcases")