https://github.com/GPflow/GPflow
Tip revision: 3065dee5fed25d5dd06692be470244ecf260cb20 authored by Mark van der Wilk on 16 August 2017, 09:00:37 UTC
Remove pandas (#486)
Remove pandas (#486)
Tip revision: 3065dee
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()