NickRF
diff --git a/‎lib/matplotlib/tests/test_backend_nbagg.py
Copy file name to clipboard
+31Lines changed: 31 additions & 0 deletions b/‎lib/matplotlib/tests/test_backend_nbagg.py
Copy file name to clipboard
+31Lines changed: 31 additions & 0 deletions
@@ -0,0 +1,31 @@
+
+import os
+import subprocess
+import tempfile
+import nbformat
+
+
+# From https://blog.thedataincubator.com/2016/06/testing-jupyter-notebooks/
+
+def _notebook_run(nb_file):
+    """Execute a notebook via nbconvert and collect output.
+       :returns (parsed nb object, execution errors)
+    """
+    with tempfile.NamedTemporaryFile(suffix=".ipynb") as fout:
+        args = ["jupyter", "nbconvert", "--to", "notebook", "--execute",
+          "--ExecutePreprocessor.timeout=500",
+          "--output", fout.name, nb_file]
+        subprocess.check_call(args)
+
+        fout.seek(0)
+        nb = nbformat.read(fout, nbformat.current_nbformat)
+
+    errors = [output for cell in nb.cells if "outputs" in cell
+                     for output in cell["outputs"]
+                     if output.output_type == "error"]
+    return nb, errors
+
+
+def test_ipynb():
+    nb, errors = _notebook_run('lib/matplotlib/tests/test_nbagg_01.ipynb')
+    assert errors == []