Customizable run of ipython notebook from command line. Requires runipy (https://github.com/paulgb/runipy) library

test_ipynb.py

#!/usr/bin/env python

import argparse

from runipy.notebook_runner import NotebookRunner
from IPython.nbformat.current import read

parser = argparse.ArgumentParser()

parser.add_argument('-n', '--notebook',
                    type=str, required=True,
                    help='notebook to run')
parser.add_argument('-p', '--pylab',
                    action='store_true',
                    help='enable pylab')
parser.add_argument('-w', '--write',
                    action='store_true',
                    help='write result back')

args = parser.parse_args()
# add extension if not given
notebook = '{name}{ext}'.format(name=args.notebook, ext=''
                                if '.' in args.notebook else '.ipynb')

print('Checking: {}'.format(notebook))
notebook_data = read(open(notebook), 'json')
r = NotebookRunner(notebook_data, pylab=args.pylab)
r.run_notebook()

if args.write:
    from IPython.nbformat.current import write
    write(r.nb, open(notebook, 'w'), 'json')