def test_missing_realization(ert_statoil_test_data):
path_file = "pathfile2.txt"
export_file = "export.txt"
csv_export2.csv_exporter(
runpathfile=path_file,
time_index="monthly",
outputfile=export_file,
column_keys="FOPR",
)
verifyExportedFile(export_file, test_header, {("iter-1", 2), ("iter-1", 0)})
def test_no_iterations(ert_statoil_test_data):
path_file = "pathfile1.txt"
export_file = "export.txt"
with pytest.raises(KeyError):
csv_export2.csv_exporter(
runpathfile=path_file,
time_index="monthly",
outputfile=export_file,
column_keys="FOPR",
)
def test_failed_realization_no_summary_file(ert_statoil_test_data):
path_file = "pathfile_path_to_failed_realizations.txt"
export_file = "export.txt"
csv_export2.csv_exporter(
runpathfile=path_file,
time_index="monthly",
outputfile=export_file,
column_keys="FOPR",
)
verifyExportedFile(
export_file,
test_header,
{("iter-1", 1), ("iter-2", 2), ("iter-1", 0), ("iter-1", 2)},
)
def main(args):
"""Setup parser"""
parser = argparse.ArgumentParser()
parser.add_argument("runpathfile", type=str)
parser.add_argument("outputfile", type=str)
parser.add_argument("time_index", type=str, default="monthly")
parser.add_argument("column_keys", nargs="+", default=None)
args = parser.parse_args(args)
csv_export2.csv_exporter(
runpathfile=args.runpathfile,
time_index=args.time_index,
outputfile=args.outputfile,
column_keys=args.column_keys,
)
print("{} csv-export written to {}".format(args.time_index,
args.outputfile))
def test_iterations(ert_statoil_test_data):
path_file = "pathfile3.txt"
export_file = "export.txt"
csv_export2.csv_exporter(
runpathfile=path_file,
time_index="monthly",
outputfile=export_file,
column_keys="FOPR",
)
verifyExportedFile(
export_file,
test_header,
{
("iter-1", 0),
("iter-1", 1),
("iter-1", 2),
("iter-2", 0),
("iter-2", 1),
("iter-2", 2),
},
)