def get_observations(ert, ensemble_id1, keyword: str, poly_ran):
obs = run_in_loop(
records.get_record_observations(res=ert,
ensemble_id=ensemble_id1,
name=keyword))
if "PSUM" in keyword:
n = int(keyword[4:])
if n < poly_ran["sum_obs_count"]:
count = poly_ran["summary_data_entries"] // poly_ran[
"sum_obs_every"]
assert len(obs) == 1
assert len(obs[0].errors) == count
assert len(obs[0].x_axis) == count
assert len(obs[0].values) == count
else:
assert len(obs) == 0
elif "POLY_RES_" in keyword:
n = int(keyword.split("@")[0][9:])
if n < poly_ran["gen_obs_count"]:
count = poly_ran["gen_data_entries"] // poly_ran["gen_obs_every"]
assert len(obs) == 1
assert len(obs[0].errors) == count
assert len(obs[0].x_axis) == count
assert len(obs[0].values) == count
else:
assert len(obs) == 0
else:
assert False, f"should never get here, keyword is {keyword}"
def get_result(ert, ensemble_id1, keyword, poly_ran):
csv = run_in_loop(
responses.get_ensemble_response_dataframe(res=ert,
ensemble_id=ensemble_id1,
response_name=keyword)).body
response_df1 = pd.read_csv(io.BytesIO(csv),
index_col=0,
float_precision="round_trip")
assert len(response_df1.columns) == poly_ran["gen_data_entries"]
assert len(response_df1.index) == poly_ran["reals"]
def get_record_parquet(ert, ensemble_id1, keyword, poly_ran):
parquet = run_in_loop(
records.get_ensemble_record(
res=ert,
name=keyword,
ensemble_id=ensemble_id1,
accept="application/x-parquet",
)).body
record_df1 = pd.read_parquet(io.BytesIO(parquet))
assert len(record_df1.columns) == poly_ran["gen_data_entries"]
assert len(record_df1.index) == poly_ran["reals"]
def get_single_record_csv(ert, ensemble_id1, keyword, poly_ran):
csv = run_in_loop(
records.get_ensemble_record(
res=ert,
name=keyword,
ensemble_id=ensemble_id1,
realization_index=poly_ran["reals"] - 1,
)).body
record_df1_indexed = pd.read_csv(io.BytesIO(csv),
index_col=0,
float_precision="round_trip")
assert len(record_df1_indexed.columns) == poly_ran["gen_data_entries"]
assert len(record_df1_indexed.index) == 1
def get_parameters(ert, ensemble_id1, keyword, poly_ran):
parameters_json = run_in_loop(
records.get_ensemble_parameters(res=ert, ensemble_id=ensemble_id1))
assert (len(parameters_json) == poly_ran["parameter_entries"] *
poly_ran["parameter_count"])