test mock property fix

ankona · Oct 21, 2024 · d93e15c · d93e15c
1 parent ae2446e
commit d93e15c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_generator.py b/tests/test_generator.py
@@ -105,7 +105,7 @@ def mock_job() -> unittest.mock.MagicMock:
     job = unittest.mock.MagicMock(
         **{
             "entity": EchoHelloWorldEntity(),
-            "name": "test_job",
+            "name": unittest.mock.PropertyMock(return_value="test_job"),
             "get_launch_steps": unittest.mock.MagicMock(
                 side_effect=lambda: NotImplementedError()
             ),
@@ -344,8 +344,8 @@ def test_exp_private_generate_method(
     exp = Experiment(name="experiment_name", exp_path=test_dir)
     job_paths = exp._generate(generator_instance, mock_job, mock_index)
     assert osp.isdir(job_paths.run_path)
-    assert job_paths.out_path.name == f"{mock_job.entity.name}.out"
-    assert job_paths.err_path.name == f"{mock_job.entity.name}.err"
+    assert job_paths.out_path.name == mock_job.entity.name + ".out"
+    assert job_paths.err_path.name == mock_job.entity.name + ".err"
 
 
 def test_generate_ensemble_directory_start(