def test_default_realizations(self):
config_file = self.createTestPath("local/poly_example/poly.ert")
with ErtTestContext("test_default_realizations",
config_file) as work_area:
ert = work_area.getErt()
args = Namespace(realizations=None)
ensemble_size = ert.getEnsembleSize()
res = model_factory._realizations(args, ensemble_size)
self.assertEqual([True] * 100, res)
def test_custom_realizations(self):
config_file = self.createTestPath("local/poly_example/poly.ert")
with ErtTestContext("test_custom_realizations",
config_file) as work_area:
ert = work_area.getErt()
args = Namespace(realizations="0-4,7,8")
ensemble_size = ert.getEnsembleSize()
res = model_factory._realizations(args, ensemble_size)
self.assertEqual(
ActiveRange(rangestring="0-4,7,8", length=ensemble_size).mask,
res)
def test_custom_realizations(self):
config_file = self.createTestPath('local/poly_example/poly.ert')
with ErtTestContext('test_custom_realizations', config_file) as work_area:
ert = work_area.getErt()
notifier = ErtCliNotifier(ert, config_file)
ERT.adapt(notifier)
args = Namespace(realizations="0-4,7,8")
res = model_factory._realizations(args)
ensemble_size = ERT.ert.getEnsembleSize()
mask = BoolVector(default_value=False, initial_size=ensemble_size)
mask.updateActiveMask("0-4,7,8")
self.assertEqual(mask, res)
