Add one test to check if the cmd module is working

Author: Martin Urban

.github/workflows/build_exe.yaml

@@ -62,8 +62,17 @@ jobs:
           source .venv/bin/activate
           python automations/my_automator.py setup dev-env
           python automations/my_automator.py build so
+          python automations/my_automator.py build wheel
+          pip install dist/*.whl
           python automations/my_automator.py build exe
 
+      - name: Run pytest
+        run: |
+          source .venv/bin/activate
+          pip install pytest
+          cd tests/
+          pytest
+
       - name: Upload artifact
         uses: actions/upload-artifact@v4
         with:

.github/workflows/build_wheel.yaml

@@ -62,6 +62,13 @@ jobs:
           python automations/my_automator.py build so
           python automations/my_automator.py build wheel
 
+      - name: Run pytest
+        run: |
+          source .venv/bin/activate
+          pip install pytest
+          cd tests/
+          pytest
+
       - name: Upload artifact
         uses: actions/upload-artifact@v4
         with:

automations/my_automator.py

@@ -17,6 +17,7 @@
 import build_macos_wheel
 import build_macos_exe
 import build_macos_so
+import run_pytest
 import dev_env
 
 
@@ -46,6 +47,10 @@
         "func": build_macos_so.build_cmd_module
       }
     }
+  },
+  "test": {
+    "help": "Runs all tests under the tests/ directory using pytest.",
+    "func": run_pytest.run_pytest_suite
   }
 }
 

automations/run_pytest.py

@@ -0,0 +1,12 @@
+import pathlib
+import subprocess
+
+import const
+
+
+def run_pytest_suite():
+  """Runs the pytest suite."""
+  subprocess.run(
+    [pathlib.Path(const.PROJECT_ROOT_DIR / ".venv/bin/pytest")],
+    cwd=pathlib.Path(const.PROJECT_ROOT_DIR / "tests")
+  )

tests/__init__.py

@@ -0,0 +1,26 @@
+"""
+#A* -------------------------------------------------------------------
+#B* This file contains source code for running a simple example
+#C* Copyright 2025 by Martin Urban.
+#D* -------------------------------------------------------------------
+#E* It is unlawful to modify or remove this copyright notice.
+#F* -------------------------------------------------------------------
+#G* Please see the accompanying LICENSE file for further information.
+#H* -------------------------------------------------------------------
+#I* Additional authors of this source file include:
+#-*
+#-*
+#-*
+#Z* -------------------------------------------------------------------
+"""
+import pathlib
+
+from pymol import cmd
+
+_FILEPATH = pathlib.Path(__file__).parent
+
+
+def test_fetch():
+  cmd.fetch("3bmp")
+  assert pathlib.Path(_FILEPATH / "3bmp.cif").exists() is True
+  assert cmd.get_model("3bmp").atom[0].model == "3bmp"