Add nbconvert test

rosbo · rosbo · commit ee2e7204cd29 · 2020-04-15T00:30:15.000Z
diff --git a/tests/data/notebook.ipynb b/tests/data/notebook.ipynb
@@ -0,0 +1,29 @@
+{
+  "cells": [
+    {
+      "metadata": {
+        "trusted": true
+      },
+      "cell_type": "code",
+      "source": "x=999\nprint(x)",
+      "execution_count": null,
+      "outputs": []
+    }
+  ],
+  "metadata":{
+    "language_info":{
+       "codemirror_mode":{
+          "name":"ipython",
+          "version":3
+       },
+       "file_extension":".py",
+       "mimetype":"text/x-python",
+       "name":"python",
+       "nbconvert_exporter":"python",
+       "pygments_lexer":"ipython3",
+       "version":"3.7.6"
+    }
+  },
+  "nbformat":4,
+  "nbformat_minor":4
+}
diff --git a/tests/test_jupyter_nbconvert.py b/tests/test_jupyter_nbconvert.py
@@ -0,0 +1,20 @@
+import unittest
+
+import subprocess
+
+class TestJupyterNbconvert(unittest.TestCase):
+    def test_nbconvert(self):
+        result = subprocess.run([
+            'jupyter',
+            'nbconvert',
+            '--to',
+            'notebook',
+            '--template',
+            '/opt/kaggle/nbconvert-extensions.tpl',
+            '--execute',
+            '--stdout',
+            '/input/tests/data/notebook.ipynb',
+        ], stdout=subprocess.PIPE)
+
+        self.assertEqual(0, result.returncode)
+        self.assertTrue(b'999' in result.stdout)
