Merge pull request #1067 from romainx/test_improvements

Added basic tests on some important components

scipy-notebook/test/test_pandas.py

@@ -0,0 +1,26 @@
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+import logging
+
+import pytest
+
+LOGGER = logging.getLogger(__name__)
+
+
+@pytest.mark.parametrize(
+    "name,command",
+    [
+        (
+            "Sum series",
+            "import pandas as pd; import numpy as np; np.random.seed(0); print(pd.Series(np.random.randint(0, 7, size=10)).sum())",
+        ),
+    ],
+)
+def test_pandas(container, name, command):
+    """Basic pandas tests"""
+    LOGGER.info(f"Testing pandas: {name} ...")
+    c = container.run(tty=True, command=["start.sh", "python", "-c", command])
+    rv = c.wait(timeout=30)
+    assert rv == 0 or rv["StatusCode"] == 0
+    logs = c.logs(stdout=True).decode("utf-8")
+    LOGGER.debug(logs)

tensorflow-notebook/test/test_tensorflow.py

@@ -0,0 +1,30 @@
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+import logging
+
+import pytest
+
+LOGGER = logging.getLogger(__name__)
+
+
+@pytest.mark.parametrize(
+    "name,command",
+    [
+        (
+            "Hello world",
+            "import tensorflow as tf;print(tf.constant('Hello, TensorFlow'))",
+        ),
+        (
+            "Sum",
+            "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))",
+        ),
+    ],
+)
+def test_tensorflow(container, name, command):
+    """Basic tensorflow tests"""
+    LOGGER.info(f"Testing tensorflow: {name} ...")
+    c = container.run(tty=True, command=["start.sh", "python", "-c", command])
+    rv = c.wait(timeout=30)
+    assert rv == 0 or rv["StatusCode"] == 0
+    logs = c.logs(stdout=True).decode("utf-8")
+    LOGGER.debug(logs)