diff --git a/‎Dockerfile‎
Lines changed: 2 additions & 0 deletions b/‎Dockerfile‎
Lines changed: 2 additions & 0 deletions
diff --git a/‎tests/data/notebook.ipynb‎
Lines changed: 2 additions & 1 deletion b/‎tests/data/notebook.ipynb‎
Lines changed: 2 additions & 1 deletion
diff --git a/‎tests/test_papermill.py‎
Lines changed: 14 additions & 0 deletions b/‎tests/test_papermill.py‎
Lines changed: 14 additions & 0 deletions
@@ -422,6 +422,8 @@ RUN pip install flashtext && \
     pip install geopandas==0.6.3 && \
     pip install nnabla && \
     pip install vowpalwabbit && \
+    # papermill can replace nbconvert for executing notebooks
+    pip install papermill && \
     /tmp/clean-layer.sh
 
 # Tesseract and some associated utility packages
 
@@ -11,6 +11,7 @@
     }
   ],
   "metadata":{
+    "kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},
     "language_info":{
        "codemirror_mode":{
           "name":"ipython",
@@ -26,4 +27,4 @@
   },
   "nbformat":4,
   "nbformat_minor":4
-}
+}
@@ -0,0 +1,14 @@
+import unittest
+
+import subprocess
+
+class TestPapermill(unittest.TestCase):
+    def test_papermill(self):
+        result = subprocess.run([
+            'papermill',
+            '/input/tests/data/notebook.ipynb',
+            '-',
+        ], stdout=subprocess.PIPE)
+
+        self.assertEqual(0, result.returncode)
+        self.assertTrue(b'999' in result.stdout)