def test_simulation_context(self):
with ErtTestContext("ert/server/rpc/simulation_context",
self.config1) as test_context:
ert = test_context.getErt()
size = 4
first_half = initializeCase(ert, "first_half", size)
other_half = initializeCase(ert, "other_half", size)
simulation_context = SimulationContext(ert, size)
for iens in range(size):
if iens % 2 == 0:
simulation_context.addSimulation(iens, first_half)
else:
simulation_context.addSimulation(iens, other_half)
self.assertFalse(
simulation_context.isRealizationFinished(iens))
with self.assertRaises(UserWarning):
simulation_context.addSimulation(size, first_half)
with self.assertRaises(UserWarning):
simulation_context.addSimulation(0, first_half)
while simulation_context.isRunning():
time.sleep(1.0)
self.assertEqual(simulation_context.getNumFailed(), 0)
self.assertEqual(simulation_context.getNumRunning(), 0)
self.assertEqual(simulation_context.getNumSuccess(), size)
first_half_state_map = first_half.getStateMap()
other_half_state_map = other_half.getStateMap()
for iens in range(size):
self.assertTrue(simulation_context.didRealizationSucceed(iens))
self.assertFalse(simulation_context.didRealizationFail(iens))
self.assertTrue(simulation_context.isRealizationFinished(iens))
if iens % 2 == 0:
self.assertEqual(first_half_state_map[iens],
RealizationStateEnum.STATE_HAS_DATA)
self.assertEqual(other_half_state_map[iens],
RealizationStateEnum.STATE_INITIALIZED)
else:
self.assertEqual(first_half_state_map[iens],
RealizationStateEnum.STATE_INITIALIZED)
self.assertEqual(other_half_state_map[iens],
RealizationStateEnum.STATE_HAS_DATA)
pfx = 'SimulationContext('
self.assertEqual(pfx, repr(simulation_context)[:len(pfx)])
def test_simulation_context(self):
config_file = self.createTestPath(
"local/snake_oil_no_data/snake_oil.ert")
with ErtTestContext("ert/server/rpc/simulation_context",
config_file) as test_context:
ert = test_context.getErt()
size = 4
mask1 = BoolVector(initial_size=size)
mask2 = BoolVector(initial_size=size)
for iens_2 in range(size / 2):
mask1[2 * iens_2] = True
mask1[2 * iens_2 + 1] = False
mask2[2 * iens_2] = False
mask2[2 * iens_2 + 1] = True
fs_manager = ert.getEnkfFsManager()
first_half = fs_manager.getFileSystem("first_half")
other_half = fs_manager.getFileSystem("other_half")
simulation_context1 = SimulationContext(ert, first_half, mask1, 0)
simulation_context2 = SimulationContext(ert, other_half, mask2, 0)
ert.createRunpath(simulation_context1.get_run_context())
ert.createRunpath(simulation_context2.get_run_context())
geo_id = 0
for iens in range(size):
if iens % 2 == 0:
simulation_context1.addSimulation(iens, geo_id)
self.assertFalse(
simulation_context1.isRealizationFinished(iens))
else:
simulation_context2.addSimulation(iens, geo_id)
self.assertFalse(
simulation_context2.isRealizationFinished(iens))
with self.assertRaises(UserWarning):
simulation_context1.addSimulation(size, geo_id)
with self.assertRaises(UserWarning):
simulation_context1.addSimulation(0, geo_id)
while simulation_context1.isRunning():
time.sleep(1.0)
while simulation_context2.isRunning():
time.sleep(1.0)
self.assertEqual(simulation_context1.getNumFailed(), 0)
self.assertEqual(simulation_context1.getNumRunning(), 0)
self.assertEqual(simulation_context1.getNumSuccess(), size / 2)
self.assertEqual(simulation_context2.getNumFailed(), 0)
self.assertEqual(simulation_context2.getNumRunning(), 0)
self.assertEqual(simulation_context2.getNumSuccess(), size / 2)
first_half_state_map = first_half.getStateMap()
other_half_state_map = other_half.getStateMap()
for iens in range(size):
if iens % 2 == 0:
self.assertTrue(
simulation_context1.didRealizationSucceed(iens))
self.assertFalse(
simulation_context1.didRealizationFail(iens))
self.assertTrue(
simulation_context1.isRealizationFinished(iens))
self.assertEqual(first_half_state_map[iens],
RealizationStateEnum.STATE_HAS_DATA)
else:
self.assertTrue(
simulation_context2.didRealizationSucceed(iens))
self.assertFalse(
simulation_context2.didRealizationFail(iens))
self.assertTrue(
simulation_context2.isRealizationFinished(iens))
self.assertEqual(other_half_state_map[iens],
RealizationStateEnum.STATE_HAS_DATA)