hpc-runner 0.3.1__tar.gz → 0.3.2__tar.gz

{hpc_runner-0.3.1 → hpc_runner-0.3.2}/.github/workflows/ci.yml

@@ -47,6 +47,9 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
+      - name: Install environment-modules
+        run: sudo apt-get update && sudo apt-get install -y environment-modules
+
       - name: Install dependencies
         run: pip install -e ".[dev]"
 

{hpc_runner-0.3.1 → hpc_runner-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hpc-runner
-Version: 0.3.1
+Version: 0.3.2
 Summary: Unified HPC job submission across multiple schedulers
 Project-URL: Homepage, https://github.com/sjalloq/hpc-runner
 Project-URL: Repository, https://github.com/sjalloq/hpc-runner

{hpc_runner-0.3.1 → hpc_runner-0.3.2}/src/hpc_runner/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.3.1'
-__version_tuple__ = version_tuple = (0, 3, 1)
+__version__ = version = '0.3.2'
+__version_tuple__ = version_tuple = (0, 3, 2)
 
 __commit_id__ = commit_id = None

hpc_runner-0.3.2/tests/modulefiles/dummy_lib/2.5

@@ -0,0 +1,4 @@
+#%Module
+module-whatis "Dummy library v2.5 for testing"
+setenv DUMMY_LIB_VERSION 2.5
+setenv DUMMY_LIB_LOADED true

hpc_runner-0.3.2/tests/modulefiles/dummy_tool/1.0

@@ -0,0 +1,4 @@
+#%Module
+module-whatis "Dummy tool v1.0 for testing"
+setenv DUMMY_TOOL_VERSION 1.0
+setenv DUMMY_TOOL_LOADED true

{hpc_runner-0.3.1 → hpc_runner-0.3.2}/tests/test_schedulers/test_local.py

@@ -1,6 +1,6 @@
 """Tests for local scheduler."""
 
-import time
+import shutil
 from pathlib import Path
 
 import pytest
@@ -9,6 +9,15 @@ from hpc_runner.core.job import Job
 from hpc_runner.core.result import JobStatus
 from hpc_runner.schedulers.local import LocalScheduler
 
+MODULEFILES_DIR = str(Path(__file__).parent.parent / "modulefiles")
+
+has_modules = (
+    shutil.which("modulecmd") is not None
+    or Path("/usr/share/Modules/init/bash").exists()
+    or Path("/etc/profile.d/modules.sh").exists()
+)
+requires_modules = pytest.mark.skipif(not has_modules, reason="Environment Modules not installed")
+
 
 class TestLocalScheduler:
     """Tests for LocalScheduler."""
@@ -151,3 +160,60 @@ class TestLocalScheduler:
         assert "module use /opt/modulefiles" in script
         assert "module load python/3.11" in script
         assert "module load gcc/12.2" in script
+
+    @requires_modules
+    def test_module_loading(self, temp_dir):
+        """Test that a module is actually loaded and sets env vars."""
+        scheduler = LocalScheduler()
+        job = Job(
+            command="echo $DUMMY_TOOL_LOADED",
+            modules=["dummy_tool/1.0"],
+            modules_path=[MODULEFILES_DIR],
+            workdir=temp_dir,
+            stdout="out.log",
+        )
+
+        result = scheduler.submit(job, interactive=True)
+
+        assert result.returncode == 0
+        stdout_path = scheduler.get_output_path(result.job_id, "stdout")
+        content = stdout_path.read_text().strip()
+        assert content == "true"
+
+    @requires_modules
+    def test_multiple_modules(self, temp_dir):
+        """Test loading multiple modules sets all env vars."""
+        scheduler = LocalScheduler()
+        job = Job(
+            command="echo $DUMMY_TOOL_LOADED $DUMMY_LIB_LOADED",
+            modules=["dummy_tool/1.0", "dummy_lib/2.5"],
+            modules_path=[MODULEFILES_DIR],
+            workdir=temp_dir,
+            stdout="out.log",
+        )
+
+        result = scheduler.submit(job, interactive=True)
+
+        assert result.returncode == 0
+        stdout_path = scheduler.get_output_path(result.job_id, "stdout")
+        content = stdout_path.read_text().strip()
+        assert content == "true true"
+
+    @requires_modules
+    def test_module_sets_env_var(self, temp_dir):
+        """Test that module sets specific version string in environment."""
+        scheduler = LocalScheduler()
+        job = Job(
+            command="echo $DUMMY_TOOL_VERSION",
+            modules=["dummy_tool/1.0"],
+            modules_path=[MODULEFILES_DIR],
+            workdir=temp_dir,
+            stdout="out.log",
+        )
+
+        result = scheduler.submit(job, interactive=True)
+
+        assert result.returncode == 0
+        stdout_path = scheduler.get_output_path(result.job_id, "stdout")
+        content = stdout_path.read_text().strip()
+        assert content == "1.0"