Fix sim context tests by using the correct queue API

src/ert/simulator/batch_simulator_context.py

@@ -55,11 +55,11 @@ def status(self) -> Status:
         Will return the state of the simulations.
         """
         return Status(
-            running=self._job_queue.count_status(RealizationState.RUNNING),
-            waiting=self._job_queue.count_status(RealizationState.WAITING),
-            pending=self._job_queue.count_status(RealizationState.PENDING),
-            complete=self._job_queue.count_status(RealizationState.SUCCESS),
-            failed=self._job_queue.count_status(RealizationState.FAILED),
+            running=self.job_queue.count_realization_state(RealizationState.RUNNING),
+            waiting=self.job_queue.count_realization_state(RealizationState.WAITING),
+            pending=self.job_queue.count_realization_state(RealizationState.PENDING),
+            complete=self.job_queue.count_realization_state(RealizationState.SUCCESS),
+            failed=self.job_queue.count_realization_state(RealizationState.FAILED),
         )
 
     def results(self) -> List[Optional[Dict[str, "npt.NDArray[np.float64]"]]]:

src/ert/simulator/simulation_context.py

@@ -54,7 +54,7 @@
     for index, run_arg in enumerate(run_context):
         if not run_context.is_active(index):
             continue
-        job_queue.add_job_from_run_arg(
+        job_queue.add_realization_from_run_arg(
             run_arg,
             ert.ert_config.queue_config.job_script,
             max_runtime,
@@ -68,12 +68,12 @@
     ):
         queue_evaluators = [
             partial(
                 job_queue.stop_long_running_jobs,
                 ert.ert_config.analysis_config.minimum_required_realizations,
             )
         ]
 
     asyncio.run(job_queue.execute(evaluators=queue_evaluators))
 
     run_context.sim_fs.sync()
 
@@ -156,7 +156,7 @@
         queue_index = self.get_run_args(iens).queue_index
         if queue_index is None:
             raise ValueError("Queue index not set")
         return self.job_queue.realization_state(queue_index) == RealizationState.SUCCESS
 
     def didRealizationFail(self, iens: int) -> bool:
         # For the purposes of this class, a failure should be anything (killed
@@ -170,7 +170,7 @@
         if queue_index is not None:
             return not (
                 self.job_queue.realization_state(queue_index)
                 in [RealizationState.SUCCESS, RealizationState.WAITING]
             )
         else:
             # job was not submitted
@@ -182,10 +182,10 @@
 
     def __repr__(self) -> str:
         running = "running" if self.isRunning() else "not running"
-        numRunn = self.job_queue.count_status(RealizationState.RUNNING)
-        numSucc = self.job_queue.count_status(RealizationState.SUCCESS)
-        numFail = self.job_queue.count_status(RealizationState.FAILED)
-        numWait = self.job_queue.count_status(RealizationState.WAITING)
+        numRunn = self.job_queue.count_realization_state(RealizationState.RUNNING)
+        numSucc = self.job_queue.count_realization_state(RealizationState.SUCCESS)
+        numFail = self.job_queue.count_realization_state(RealizationState.FAILED)
+        numWait = self.job_queue.count_realization_state(RealizationState.WAITING)
         return (
             f"SimulationContext({running}, #running = {numRunn}, "
             f"#success = {numSucc}, #failed = {numFail}, #waiting = {numWait})"
@@ -226,7 +226,7 @@
         if queue_index is None:
             # job was not submitted
             return None
         if self.job_queue.realization_state(queue_index) == RealizationState.WAITING:
             return None
 
         return ForwardModelStatus.load(run_arg.runpath)

tests/unit_tests/simulator/test_batch_sim.py

@@ -5,7 +5,7 @@
 import pytest
 
 from ert.config import ErtConfig
-from ert.job_queue import JobStatus
+from ert.realization_state import RealizationState
 from ert.simulator import BatchSimulator
 
 
@@ -409,13 +409,13 @@ def test_stop_sim(copy_case, storage):
 def assertContextStatusOddFailures(batch_ctx, final_state_only=False):
     running_status = set(
         (
-            JobStatus.WAITING,
-            JobStatus.SUBMITTED,
-            JobStatus.PENDING,
-            JobStatus.RUNNING,
-            JobStatus.UNKNOWN,
-            JobStatus.EXIT,
-            JobStatus.DONE,
+            RealizationState.WAITING,
+            RealizationState.SUBMITTED,
+            RealizationState.PENDING,
+            RealizationState.RUNNING,
+            RealizationState.UNKNOWN,
+            RealizationState.EXIT,
+            RealizationState.DONE,
             None,  # job is not submitted yet but ok for this test
         )
     )
@@ -425,9 +425,9 @@ def assertContextStatusOddFailures(batch_ctx, final_state_only=False):
         if not final_state_only and status in running_status:
             continue
         if idx % 2 == 0:
-            assert status == JobStatus.SUCCESS
+            assert status == RealizationState.SUCCESS
         else:
-            assert status == JobStatus.FAILED
+            assert status == RealizationState.FAILED
 
 
 def test_batch_ctx_status_failing_jobs(setup_case, storage):

tests/unit_tests/simulator/test_simulation_context.py

@@ -46,13 +46,17 @@ def test_simulation_context(setup_case, storage):
                 f"runpath/realization-{iens}-{iens}/iter-0"
             )
 
-    assert even_ctx.job_queue.count_status(RealizationState.FAILED) == 0
-    assert even_ctx.job_queue.count_status(RealizationState.RUNNING) == 0
-    assert even_ctx.job_queue.count_status(RealizationState.SUCCESS) == size / 2
+    assert even_ctx.job_queue.count_realization_state(RealizationState.FAILED) == 0
+    assert even_ctx.job_queue.count_realization_state(RealizationState.RUNNING) == 0
+    assert (
+        even_ctx.job_queue.count_realization_state(RealizationState.SUCCESS) == size / 2
+    )
 
-    assert odd_ctx.job_queue.count_status(RealizationState.FAILED) == 0
-    assert odd_ctx.job_queue.count_status(RealizationState.RUNNING) == 0
-    assert odd_ctx.job_queue.count_status(RealizationState.SUCCESS) == size / 2
+    assert odd_ctx.job_queue.count_realization_state(RealizationState.FAILED) == 0
+    assert odd_ctx.job_queue.count_realization_state(RealizationState.RUNNING) == 0
+    assert (
+        odd_ctx.job_queue.count_realization_state(RealizationState.SUCCESS) == size / 2
+    )
 
     for iens in range(size):
         if iens % 2 == 0: