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):
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)])
