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)