wip: test num_cpu through baserunmodel

equinor · Jun 17, 2024 · 7f3434d · 7f3434d
1 parent 4ebe463
commit 7f3434d
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/unit_tests/run_models/test_base_run_model.py b/tests/unit_tests/run_models/test_base_run_model.py
@@ -180,3 +180,19 @@ def test_delete_run_path(run_path_format, active_realizations):
     assert all(path.parent.exists() for path in expected_removed)
     assert all(path.exists() for path in expected_remaining)
     assert share_path.exists()
+
+
+def test_num_cpu_is_propagated_from_config_to_ensemble(create_dummy_run_path):
+    config = MagicMock()
+    config.model_config = ModelConfig(runpath_format_string="realization-<IENS>")
+    config.substitution_list = SubstitutionList()
+
+    simulation_arguments = EnsembleExperimentRunArguments(
+        active_realizations=[0],
+        current_ensemble="Case_Name",
+        minimum_required_realizations=0,
+        ensemble_size=1,
+        stop_long_running=False,
+    )
+
+    brm = BaseRunModel(simulation_arguments, config, None, None, None)