Use the IPython ExecutePreprocessor in tests

sigmavirus24 · sigmavirus24 · commit 8f0962378050 · 2015-06-23T22:18:41.000-05:00
diff --git a/tests/nbtest.py b/tests/nbtest.py
@@ -2,16 +2,23 @@
 # -*- coding: utf-8 -*-
 
 import glob
+import json
 
 from IPython.config import Config
+from IPython.nbconvert import ExecutePreprocessor
 from IPython.nbconvert import HTMLExporter
+from IPython.nbformat import NotebookNode
 
 c = Config({
-  'ExecutePreprocessor': {'enabled': True}
+    'ExecutePreprocessor': {'enabled': True}
 })
 
 exporter = HTMLExporter(config=c)
+executor = ExecutePreprocessor(config=c)
 
 for filename in glob.glob("example-notebooks/*.ipynb"):
-  print(filename)
-  exporter.from_filename(filename)
+    print(filename)
+    exporter.from_filename(filename)
+    with open(filename, 'r') as file:
+        node = NotebookNode(json.load(file))
+    executor.preprocess(node, {})
