def test_arg_loader(self):
with self.assertRaises(IOError):
arg = ArgLoader.load("arg1X")
arg_file = self.createTestPath(
"Statoil/config/with_GEN_DATA_RFT/wellpath/WI_1.txt")
with self.assertRaises(ValueError):
arg = ArgLoader.load(
arg_file,
column_names=["Col1", "Col2", "Col3", "COl5", "Col6"])
arg = ArgLoader.load(arg_file,
column_names=["utm_x", "utm_y", "md", "tvd"])
self.assertFloatEqual(arg["utm_x"][0], 461317.620646)
obs_vector = enkf_obs[obs_key]
data_key = obs_vector.getDataKey()
report_step = obs_vector.activeStep()
obs_node = obs_vector.getNode(report_step)
rft_data = GenDataCollector.loadGenData(self.ert(), case, data_key, report_step)
fs = self.ert().getEnkfFsManager().getFileSystem(case)
realizations = fs.realizationList(RealizationStateEnum.STATE_HAS_DATA)
# Trajectory
trajectory_file = os.path.join(trajectory_path, "%s.txt" % well)
if not os.path.isfile(trajectory_file):
trajectory_file = os.path.join(trajectory_path, "%s_R.txt" % well)
trajectory = WellTrajectory(trajectory_file)
arg = ArgLoader.load(trajectory_file, column_names=["utm_x", "utm_y", "md", "tvd"])
tvd_arg = arg["tvd"]
data_size = len(tvd_arg)
# Observations
obs = numpy.empty(shape=(data_size, 2), dtype=numpy.float64)
obs.fill(numpy.nan)
for obs_index in range(len(obs_node)):
data_index = obs_node.getDataIndex(obs_index)
value = obs_node.getValue(obs_index)
std = obs_node.getStandardDeviation(obs_index)
obs[data_index, 0] = value
obs[data_index, 1] = std
for iens in realizations:
realization_frame = pandas.DataFrame(data={"TVD": tvd_arg,
