def checkBlockData(self, ert, obs_key, report_step):
"""
@type ert: EnKFMain
@type obs_key: str
@type report_step: int
"""
enkf_obs = ert.getObservations()
obs_vector = enkf_obs[obs_key]
loader = PlotBlockDataLoader(obs_vector)
fs = ert.getEnkfFsManager().getCurrentFileSystem()
plot_block_data = loader.load(fs, report_step)
self.assertEqual(ert.getEnsembleSize(), len(plot_block_data))
depth = plot_block_data.getDepth()
depth_test_values = [1752.24998474, 1757.88926697, 1760.70924377]
if report_step == 56:
depth_test_values.append(1763.52885437)
self.assertAlmostEqualList(depth_test_values, depth)
block_obs = len(obs_vector.getNode(report_step))
self.assertEqual(block_obs, len(plot_block_data[0]))
self.assertEqual(block_obs, len(plot_block_data[9]))
if report_step == 50:
rft_values = [244.681655884, 245.217041016, 245.48500061]
else:
rft_values = [
239.7550354, 240.290313721, 240.558197021, 240.825881958
]
self.assertAlmostEqualList(rft_values, plot_block_data[0])
if report_step == 50:
rft_values = [238.702560425, 239.237838745, 239.505737305]
else:
rft_values = [
234.41583252, 234.95098877, 235.218841553, 235.486480713
]
self.assertAlmostEqualList(rft_values, plot_block_data[9])
def checkBlockData(self, ert, obs_key, report_step):
"""
@type ert: EnKFMain
@type obs_key: str
@type report_step: int
"""
enkf_obs = ert.getObservations()
obs_vector = enkf_obs[obs_key]
loader = PlotBlockDataLoader(obs_vector)
fs = ert.getEnkfFsManager().getCurrentFileSystem()
plot_block_data = loader.load(fs, report_step)
self.assertEqual(ert.getEnsembleSize(), len(plot_block_data))
depth = plot_block_data.getDepth()
depth_test_values = [1752.24998474, 1757.88926697, 1760.70924377]
if report_step == 56:
depth_test_values.append(1763.52885437)
self.assertAlmostEqualList(depth_test_values, depth)
block_obs = len(obs_vector.getNode(report_step))
self.assertEqual(block_obs, len(plot_block_data[0]))
self.assertEqual(block_obs, len(plot_block_data[9]))
if report_step == 50:
rft_values = [244.681655884, 245.217041016, 245.48500061]
else:
rft_values = [239.7550354, 240.290313721, 240.558197021, 240.825881958]
self.assertAlmostEqualList(rft_values, plot_block_data[0])
if report_step == 50:
rft_values = [238.702560425, 239.237838745, 239.505737305]
else:
rft_values = [234.41583252, 234.95098877, 235.218841553, 235.486480713]
self.assertAlmostEqualList(rft_values, plot_block_data[9])
def fetchData(self, key, case=None):
enkf_fs = self.ert().getEnkfFsManager().getFileSystem(case)
observations = self.ert().getObservations()
assert observations.hasKey(key)
observation_vector = observations[key]
loader = PlotBlockDataLoader(observation_vector)
report_step_data = []
for report_step in observation_vector:
block_data = loader.load(enkf_fs, report_step)
data = self.__fetchSimulationData(block_data)
data["report_step"] = report_step
report_step_data.append(data)
if self.__selected_report_step_index is not None:
return report_step_data[self.__selected_report_step_index]
else:
return report_step_data
