test mock object tweak for property string fix

CrayLabs · Oct 21, 2024 · 78a3a33 · 78a3a33
1 parent ae2446e
commit 78a3a33
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/tests/test_generator.py b/tests/test_generator.py
@@ -103,13 +103,11 @@ def files():
 def mock_job() -> unittest.mock.MagicMock:
     """Fixture to create a mock Job."""
     job = unittest.mock.MagicMock(
-        **{
-            "entity": EchoHelloWorldEntity(),
-            "name": "test_job",
-            "get_launch_steps": unittest.mock.MagicMock(
-                side_effect=lambda: NotImplementedError()
-            ),
-        },
+        get_launch_steps=unittest.mock.MagicMock(
+            side_effect=lambda: NotImplementedError()
+        ),
+        entity=EchoHelloWorldEntity(),
+        name="test_job",
         spec=Job,
     )
     yield job