Merge pull request #163 from geometric-intelligence/ninamiolane-test-…

…tutorials

Add unit tests for tutorials

tests/test_tutorials.py

@@ -0,0 +1,33 @@
+"""Unit tests for the tutorials."""
+
+import subprocess
+import tempfile
+
+import pytest
+
+
+def _exec_tutorial(path):
+    file_name = tempfile.NamedTemporaryFile(suffix=".ipynb").name
+    args = [
+        "jupyter",
+        "nbconvert",
+        "--to",
+        "notebook",
+        "--execute",
+        "--ExecutePreprocessor.timeout=1000",
+        "--ExecutePreprocessor.kernel_name=python3",
+        "--output",
+        file_name,
+        path,
+    ]
+    subprocess.check_call(args)
+
+
+TUTORIALS_DIR = "tutorials"
+paths = [f"{TUTORIALS_DIR}/01_methods_create_synthetic_data.ipynb"]
+
+
+@pytest.mark.parametrize("path", paths)
+def test_tutorial(path):
+    """Test the tutorials."""
+    _exec_tutorial(path)