Fix project check test

Matgenix · Sep 25, 2024 · 8637ef5 · 8637ef5
1 parent c146999
commit 8637ef5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/integration/conftest.py b/tests/integration/conftest.py
@@ -337,7 +337,7 @@ def write_tmp_settings(
         with open(tmp_dir / f"{random_project_name}.json", "w") as f:
             f.write(project_json)
 
-        yield
+        yield project
     finally:
         shutil.rmtree(tmp_dir)
         # Reset environment variables if they were set elsewhere

diff --git a/tests/integration/test_workers.py b/tests/integration/test_workers.py
@@ -10,14 +10,14 @@
 WORKERS = ["test_local_worker", "test_remote_slurm_worker", "test_remote_sge_worker"]
 
 
-def test_project_init(random_project_name) -> None:
+def test_project_init(random_project_name, write_tmp_settings) -> None:
     from jobflow_remote.config import ConfigManager
 
     cm = ConfigManager()
     assert len(cm.projects) == 1
     assert cm.projects[random_project_name]
     project = cm.get_project()
-    assert len(project.workers) == 4
+    assert len(project.workers) == len(write_tmp_settings.workers)
 
 
 def test_paramiko_ssh_connection(job_controller, slurm_ssh_port) -> None: