def load_active_masks(self, case1, case2): with ErtTestContext("gen_data_config_test", self.config_file) as test_context: ert = test_context.getErt() fs1 = ert.getEnkfFsManager().getFileSystem(case1) config_node = ert.ensembleConfig().getNode("TIMESHIFT") data_node = EnkfNode(config_node) data_node.tryLoad(fs1, NodeId(60, 0)) active_mask = self._get_active_mask( config_node.getDataModelConfig()) first_active_mask_length = len(active_mask) self.assertEqual(first_active_mask_length, 2560) fs2 = ert.getEnkfFsManager().getFileSystem(case2) data_node = EnkfNode(config_node) data_node.tryLoad(fs2, NodeId(60, 0)) active_mask = self._get_active_mask( config_node.getDataModelConfig()) second_active_mask_len = len(active_mask) self.assertEqual(second_active_mask_len, 2560) self.assertEqual(first_active_mask_length, second_active_mask_len) # Setting one element to False, load different case, check, reload, and check. self.assertTrue(active_mask[10]) active_mask_modified = active_mask.copy() active_mask_modified[10] = False self.updateMask(config_node.getDataModelConfig(), 60, fs2, active_mask_modified) active_mask = self._get_active_mask( config_node.getDataModelConfig()) self.assertFalse(active_mask[10]) #Load first - check element is true data_node = EnkfNode(config_node) data_node.tryLoad(fs1, NodeId(60, 0)) active_mask = self._get_active_mask( config_node.getDataModelConfig()) self.assertTrue(active_mask[10]) # Reload second again, should now be false at 10, due to the update further up data_node = EnkfNode(config_node) data_node.tryLoad(fs2, NodeId(60, 0)) active_mask = self._get_active_mask( config_node.getDataModelConfig()) self.assertFalse(active_mask[10])
def load_active_masks(self, case1, case2): with ErtTestContext("gen_data_config_test", self.config_file) as test_context: ert = test_context.getErt() fs1 = ert.getEnkfFsManager().getFileSystem(case1) config_node = ert.ensembleConfig().getNode("TIMESHIFT") data_node = EnkfNode(config_node) data_node.tryLoad(fs1, NodeId(60, 0)) active_mask = get_active_mask(config_node.getDataModelConfig()) first_active_mask_length = len(active_mask) self.assertEqual(first_active_mask_length, 2560) fs2 = ert.getEnkfFsManager().getFileSystem(case2) data_node = EnkfNode(config_node) data_node.tryLoad(fs2, NodeId(60, 0)) active_mask = get_active_mask(config_node.getDataModelConfig()) second_active_mask_len = len(active_mask) self.assertEqual(second_active_mask_len, 2560) self.assertEqual(first_active_mask_length, second_active_mask_len) # Setting one element to False, load different case, check, reload, and check. self.assertTrue(active_mask[10]) active_mask_modified = active_mask.copy() active_mask_modified[10] = False updateMask(config_node.getDataModelConfig(), 60, fs2, active_mask_modified) active_mask = get_active_mask(config_node.getDataModelConfig()) self.assertFalse(active_mask[10]) # Load first - check element is true data_node = EnkfNode(config_node) data_node.tryLoad(fs1, NodeId(60, 0)) active_mask = get_active_mask(config_node.getDataModelConfig()) self.assertTrue(active_mask[10]) # Reload second again, should now be false at 10, due to the update further up data_node = EnkfNode(config_node) data_node.tryLoad(fs2, NodeId(60, 0)) active_mask = get_active_mask(config_node.getDataModelConfig()) self.assertFalse(active_mask[10])
def test_create(self): with ErtTestContext("gen_data_test", self.config_file) as test_context: ert = test_context.getErt() fs1 = ert.getEnkfFsManager().getCurrentFileSystem() config_node = ert.ensembleConfig().getNode("TIMESHIFT") data_node = EnkfNode(config_node) data_node.tryLoad(fs1, NodeId(60, 0)) gen_data = data_node.asGenData() data = gen_data.getData() self.assertEqual(len(data), 2560)
def test_create(self): with ErtTestContext("gen_data_test", self.config_file) as test_context: ert = test_context.getErt() fs1 = ert.getEnkfFsManager().getCurrentFileSystem() config_node = ert.ensembleConfig().getNode("TIMESHIFT") data_node = EnkfNode(config_node) data_node.tryLoad(fs1, NodeId(60, 0)) gen_data = data_node.asGenData() data = gen_data.getData() self.assertEqual(len(data) , 2560)
def test_it(self): state_size = 10 with ErtTestContext("update" , self.config_file) as tc: analysis = self.createAnalysisModule() ert = tc.getErt() obs = ert.getObservations() local_obsdata = obs.getAllActiveLocalObsdata( ) fs = ert.getEnkfFsManager().getCurrentFileSystem() mask = BoolVector( initial_size = ert.getEnsembleSize() , default_value = True) meas_data = MeasData(mask) obs_data = ObsData() obs.getObservationAndMeasureData( fs , local_obsdata , mask.createActiveList() , meas_data , obs_data ) update( self.rng , mask , analysis , ert , meas_data , obs_data , state_size) mask[0] = False mask[4] = False meas_data = MeasData(mask) obs_data = ObsData() obs.getObservationAndMeasureData( fs , local_obsdata , mask.createActiveList() , meas_data , obs_data ) update( self.rng , mask , analysis , ert , meas_data , obs_data , state_size)