def test_setup_multiple_data_assimilation(self):
config_file = self.createTestPath("local/poly_example/poly.ert")
with ErtTestContext("test_single_test_run", config_file) as work_area:
ert = work_area.getErt()
facade = LibresFacade(ert)
args = Namespace(
realizations="0-4,7,8",
weights="6,4,2",
target_case="test_case_%d",
start_iteration="0",
)
model = model_factory._setup_multiple_data_assimilation(
ert,
args,
facade.get_ensemble_size(),
facade.get_current_case_name(),
)
self.assertTrue(isinstance(model, MultipleDataAssimilation))
self.assertEqual(5, len(model._simulation_arguments.keys()))
self.assertTrue(
"active_realizations" in model._simulation_arguments)
self.assertTrue("target_case" in model._simulation_arguments)
self.assertTrue("analysis_module" in model._simulation_arguments)
self.assertTrue("weights" in model._simulation_arguments)
self.assertTrue("start_iteration" in model._simulation_arguments)
model.create_context(0)
def test_setup_multiple_data_assimilation(self):
config_file = self.createTestPath('local/poly_example/poly.ert')
with ErtTestContext('test_single_test_run', config_file) as work_area:
ert = work_area.getErt()
notifier = ErtCliNotifier(ert, config_file)
ERT.adapt(notifier)
args = Namespace(realizations="0-4,7,8", weights="6,4,2", target_case="test_case")
model, argument = model_factory._setup_multiple_data_assimilation(args)
self.assertTrue(isinstance(model, MultipleDataAssimilation))
self.assertEqual(4, len(argument.keys()))
self.assertTrue("active_realizations" in argument)
self.assertTrue("target_case" in argument)
self.assertTrue("analysis_module" in argument)
self.assertTrue("weights" in argument)