def test_obs_block_scale_std(self):
with ErtTestContext("obs_test_scale",
self.config_file) as test_context:
ert = test_context.getErt()
fs = ert.getEnkfFsManager().getCurrentFileSystem()
active_list = IntVector()
active_list.initRange(0, ert.getEnsembleSize(), 1)
obs = ert.getObservations()
obs_data = LocalObsdata("OBSxx", obs)
obs_vector = obs["WWCT:OP_1"]
obs_data.addObsVector(obs_vector)
scale_factor = obs.scaleCorrelatedStd(fs, obs_data, active_list)
for obs_node in obs_vector:
for index in range(len(obs_node)):
self.assertEqual(scale_factor,
obs_node.getStdScaling(index))
def test_scale_obs(self):
with ErtTestContext("obs_test", self.config_file) as test_context:
ert = test_context.getErt()
obs = ert.getObservations()
obs1 = obs["WWCT:OP_1"].getNode(50)
obs2 = obs["WWCT:OP_1_50"].getNode(50)
self.assertEqual(obs1.getStandardDeviation(),
obs2.getStandardDeviation())
std0 = obs1.getStandardDeviation()
local_obsdata = LocalObsdata("obs", obs)
node1 = local_obsdata.addNode("WWCT:OP_1")
node2 = local_obsdata.addNode("WWCT:OP_1_50")
node1.addTimeStep(50)
node2.addTimeStep(50)
mask = BoolVector(default_value=True)
mask[2] = True
meas_data = MeasData(mask)
obs_data = ObsData()
fs = ert.getEnkfFsManager().getCurrentFileSystem()
active_list = IntVector()
active_list.initRange(0, 2, 1)
obs.getObservationAndMeasureData(fs, local_obsdata,
EnkfStateType.FORECAST,
active_list, meas_data, obs_data)
self.assertEqual(2, len(obs_data))
v1 = obs_data[0]
v2 = obs_data[1]
self.assertEqual(v1[1], std0)
self.assertEqual(v2[1], std0)
meas_data = MeasData(mask)
obs_data = ObsData(10)
obs.getObservationAndMeasureData(fs, local_obsdata,
EnkfStateType.FORECAST,
active_list, meas_data, obs_data)
self.assertEqual(2, len(obs_data))
v1 = obs_data[0]
v2 = obs_data[1]
self.assertEqual(v1[1], std0 * 10)
self.assertEqual(v2[1], std0 * 10)
actl = ActiveList()
obs1.updateStdScaling(10, actl)
obs2.updateStdScaling(20, actl)
meas_data = MeasData(mask)
obs_data = ObsData()
obs.getObservationAndMeasureData(fs, local_obsdata,
EnkfStateType.FORECAST,
active_list, meas_data, obs_data)
self.assertEqual(2, len(obs_data))
v1 = obs_data[0]
v2 = obs_data[1]
self.assertEqual(v1[1], std0 * 10)
self.assertEqual(v2[1], std0 * 20)
