From 29f09f4fb696830f9942a36a7a1bf3c56f13c350 Mon Sep 17 00:00:00 2001 From: xjules Date: Fri, 24 Nov 2023 12:59:25 +0100 Subject: [PATCH] Fix RealizationState vs RealizationStorageState imports and usage --- tests/unit_tests/simulator/test_batch_sim.py | 2 +- tests/unit_tests/simulator/test_simulation_context.py | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/unit_tests/simulator/test_batch_sim.py b/tests/unit_tests/simulator/test_batch_sim.py index dcdb24311fc..28e353e7558 100644 --- a/tests/unit_tests/simulator/test_batch_sim.py +++ b/tests/unit_tests/simulator/test_batch_sim.py @@ -5,7 +5,7 @@ import pytest from ert.config import ErtConfig -from ert.realization_state import RealizationState +from ert.job_queue import RealizationState from ert.simulator import BatchSimulator diff --git a/tests/unit_tests/simulator/test_simulation_context.py b/tests/unit_tests/simulator/test_simulation_context.py index 34fa133cfbe..854f019ceb2 100644 --- a/tests/unit_tests/simulator/test_simulation_context.py +++ b/tests/unit_tests/simulator/test_simulation_context.py @@ -1,6 +1,7 @@ from ert.enkf_main import EnKFMain -from ert.realization_state import RealizationState +from ert.job_queue import RealizationState from ert.simulator import SimulationContext +from ert.storage import RealizationStorageState from tests.utils import wait_until @@ -64,10 +65,10 @@ def test_simulation_context(setup_case, storage): assert not even_ctx.didRealizationFail(iens) assert even_ctx.isRealizationFinished(iens) - assert even_half.state_map[iens] == RealizationState.HAS_DATA + assert even_half.state_map[iens] == RealizationStorageState.HAS_DATA else: assert odd_ctx.didRealizationSucceed(iens) assert not odd_ctx.didRealizationFail(iens) assert odd_ctx.isRealizationFinished(iens) - assert odd_half.state_map[iens] == RealizationState.HAS_DATA + assert odd_half.state_map[iens] == RealizationStorageState.HAS_DATA