https://github.com/GPflow/GPflow
Tip revision: 555427747c90f4417d16da4dacab5c528cf27a64 authored by alexggmatthews on 07 August 2017, 09:22:37 UTC
Using settings context. general tidying.
Using settings context. general tidying.
Tip revision: 5554277
test_notebooks.py
from __future__ import print_function
import nbformat
from nbconvert.preprocessors import ExecutePreprocessor
from nbconvert.preprocessors.execute import CellExecutionError
import glob
import traceback
import unittest
import sys
import time
import os
from nose.plugins.attrib import attr
@attr(speed='slow')
class TestNotebooks(unittest.TestCase):
def _execNotebook(self, ep, notebook_filename, nbpath):
with open(notebook_filename) as f:
nb = nbformat.read(f, as_version=nbformat.current_nbformat)
try:
ep.preprocess(nb, {'metadata': {'path': nbpath}})
except CellExecutionError:
print('-' * 60)
traceback.print_exc(file=sys.stdout)
print('-' * 60)
self.assertTrue(False, 'Error executing the notebook %s.\
See above for error.' % notebook_filename)
def test_all_notebooks(self):
''' Test all notebooks except blacklist. Blacklisted notebooks take too long.'''
blacklist = ['svi_test.ipynb', 'GPLVM.ipynb', 'regression.ipynb']
pythonkernel = 'python'+str(sys.version_info[0])
this_dir = os.path.dirname(__file__)
nbpath = os.path.join(this_dir, '../doc/source/notebooks/')
# see http://nbconvert.readthedocs.io/en/stable/execute_api.html
ep = ExecutePreprocessor(timeout=120, kernel_name=pythonkernel, interrupt_on_timeout=True)
lfiles = glob.glob(nbpath+"*.ipynb")
for notebook_filename in lfiles:
if(os.path.basename(notebook_filename) not in blacklist):
t = time.time()
self._execNotebook(ep, notebook_filename, nbpath)
print(notebook_filename, 'took %g seconds.' % (time.time()-t))
if __name__ == '__main__':
unittest.main()