def test_multiple_parameters():
"""Test what happens when we have duplicate parameter files"""
eclfiles = EclFiles(DATAFILE)
parametersjson = Path(eclfiles.get_path()) / "parameters.json"
parameterstxt = Path(eclfiles.get_path()).parent / "parameters.txt"
parameterstxt.write_text("FOO 1\nBAR 4", encoding="utf-8")
parametersjson.write_text(json.dumps({"BAR": 5, "COM": 6}), encoding="utf-8")
param_dict = load_all(find_parameter_files(eclfiles))
assert len(param_dict) == 3
assert param_dict["BAR"] == 5 # json has precedence over txt
parametersjson.unlink()
parameterstxt.unlink()
def test_multiple_parameters():
"""Test what happens when we have duplicate parameter files"""
eclfiles = EclFiles(DATAFILE)
parametersjson = os.path.join(eclfiles.get_path(), "parameters.json")
parameterstxt = os.path.join(eclfiles.get_path(), os.pardir, "parameters.txt")
with open(parameterstxt, "w") as pfile:
pfile.write("FOO 1\nBAR 4")
with open(parametersjson, "w") as pfile:
pfile.write(json.dumps({"BAR": 5, "COM": 6}))
param_dict = load_all(find_parameter_files(eclfiles))
assert len(param_dict) == 3
assert param_dict["BAR"] == 5 # json has precedence over txt
os.unlink(parametersjson)
os.unlink(parameterstxt)
def test_paramsupport(tmpdir):
"""Test that we can merge in parameters.txt"""
tmpcsvfile = tmpdir.join(".TMP-sum.csv")
eclfiles = EclFiles(DATAFILE)
parameterstxt = os.path.join(eclfiles.get_path(), "parameters.txt")
if os.path.exists(parameterstxt):
os.remove(parameterstxt)
with open(parameterstxt, "w") as pfile:
pfile.write("FOO 1\nBAR 3")
sys.argv = ["ecl2csv", "summary", DATAFILE, "-o", str(tmpcsvfile), "-p"]
ecl2csv.main()
disk_df = pd.read_csv(tmpcsvfile)
assert "FOPT" in disk_df
assert "FOO" in disk_df
assert "BAR" in disk_df
assert disk_df["BAR"].unique()[0] == 3
os.remove(parameterstxt)
os.remove(str(tmpcsvfile))
parametersyml = os.path.join(eclfiles.get_path(), "parameters.yml")
if os.path.exists(parametersyml):
os.remove(parametersyml)
with open(parametersyml, "w") as pfile:
pfile.write(yaml.dump({"FOO": 1, "BAR": 3}))
sys.argv = ["ecl2csv", "summary", DATAFILE, "-o", str(tmpcsvfile), "-p"]
ecl2csv.main()
disk_df = pd.read_csv(str(tmpcsvfile))
assert "FOPT" in disk_df
assert "FOO" in disk_df
assert len(disk_df["FOO"].unique()) == 1
assert disk_df["FOO"].unique()[0] == 1
assert "BAR" in disk_df
assert len(disk_df["BAR"].unique()) == 1
assert disk_df["BAR"].unique()[0] == 3
os.remove(parametersyml)
def test_parameters():
"""Test import of parameters.txt++"""
eclfiles = EclFiles(DATAFILE)
# NB: This test easily fails due to remnants of other test code..
assert not find_parameter_files(eclfiles)
parameterstxt = Path(eclfiles.get_path()) / "parameters.txt"
# If this exists, it is a remnant from test code that has
# crashed. It should NOT be in git.
if parameterstxt.is_file():
parameterstxt.unlink()
parameterstxt.write_text("FOO 1\nBAR 3", encoding="utf-8")
assert Path(parameterstxt).is_file()
param_dict = load(parameterstxt)
assert "FOO" in param_dict
assert "BAR" in param_dict
assert len(find_parameter_files(eclfiles)) == 1
parameterstxt.unlink()
parameterstxt = Path(eclfiles.get_path()).parent / "parameters.txt"
if parameterstxt.is_file():
parameterstxt.unlink()
parameterstxt.write_text("FOO 1\nBAR 3\nCONTACT:BARF 2700",
encoding="utf-8")
assert Path(parameterstxt).is_file()
param_dict = load(parameterstxt)
assert "FOO" in param_dict
assert "BAR" in param_dict
assert param_dict["BAR"] == 3
assert param_dict["CONTACT:BARF"] == 2700
assert len(find_parameter_files(eclfiles)) == 1
parameterstxt.unlink()
# Typical parameters.json structure: The group "CONTACT" is assumed having
# duplicate information, and is to be ignored
dump_me = {
"FOO": 1,
"BAR": "com",
"CONTACT:BARF": 2700,
"CONTACT": {
"BARF": 2700
}
}
parametersyml = Path(eclfiles.get_path()) / "parameters.yml"
if parametersyml.is_file():
parametersyml.unlink()
parametersyml.write_text(yaml.dump(dump_me), encoding="utf-8")
assert Path(parametersyml).is_file()
assert len(find_parameter_files(eclfiles)) == 1
param_dict = load(parametersyml)
assert "FOO" in param_dict
assert "BAR" in param_dict
assert param_dict["BAR"] == "com"
parametersyml.unlink()
parametersjson = Path(eclfiles.get_path()) / "parameters.json"
if parametersjson.is_file():
parametersjson.unlink()
parametersjson.write_text(json.dumps(dump_me), encoding="utf-8")
assert Path(parametersjson).is_file()
assert len(find_parameter_files(eclfiles)) == 1
param_dict = load(find_parameter_files(eclfiles)[0])
param_dict_m = load_all(find_parameter_files(eclfiles))
assert "FOO" in param_dict
assert "BAR" in param_dict
assert param_dict["BAR"] == "com"
assert param_dict == param_dict_m
parametersjson.unlink()
def test_parameters():
"""Test import of parameters.txt++"""
eclfiles = EclFiles(DATAFILE)
# NB: This test easily fails due to remnants of other test code..
assert not find_parameter_files(eclfiles)
parameterstxt = os.path.join(eclfiles.get_path(), "parameters.txt")
if os.path.exists(parameterstxt):
# If this exists, it is a remnant from test code that has
# crashed. It should NOT be in git.
os.unlink(parameterstxt)
with open(parameterstxt, "w") as pfile:
pfile.write("FOO 1\nBAR 3")
assert os.path.exists(parameterstxt)
param_dict = load(parameterstxt)
assert "FOO" in param_dict
assert "BAR" in param_dict
assert len(find_parameter_files(eclfiles)) == 1
os.unlink(parameterstxt)
parameterstxt = os.path.join(eclfiles.get_path(), os.pardir, "parameters.txt")
if os.path.exists(parameterstxt):
os.unlink(parameterstxt)
with open(parameterstxt, "w") as pfile:
pfile.write("FOO 1\nBAR 3\n")
pfile.write("CONTACT:BARF 2700")
assert os.path.exists(parameterstxt)
param_dict = load(parameterstxt)
assert "FOO" in param_dict
assert "BAR" in param_dict
assert param_dict["BAR"] == 3
assert param_dict["CONTACT:BARF"] == 2700
assert len(find_parameter_files(eclfiles)) == 1
os.unlink(parameterstxt)
# Typical parameters.json structure: The group "CONTACT" is assumed having
# duplicate information, and is to be ignored
dump_me = {"FOO": 1, "BAR": "com", "CONTACT:BARF": 2700, "CONTACT": {"BARF": 2700}}
parametersyml = os.path.join(eclfiles.get_path(), "parameters.yml")
if os.path.exists(parametersyml):
os.unlink(parametersyml)
with open(parametersyml, "w") as pfile:
pfile.write(yaml.dump(dump_me))
assert os.path.exists(parametersyml)
assert len(find_parameter_files(eclfiles)) == 1
param_dict = load(parametersyml)
assert "FOO" in param_dict
assert "BAR" in param_dict
assert param_dict["BAR"] == "com"
os.unlink(parametersyml)
parametersjson = os.path.join(eclfiles.get_path(), "parameters.json")
if os.path.exists(parametersjson):
os.unlink(parametersjson)
with open(parametersjson, "w") as pfile:
pfile.write(json.dumps(dump_me))
assert os.path.exists(parametersjson)
assert len(find_parameter_files(eclfiles)) == 1
param_dict = load(find_parameter_files(eclfiles)[0])
param_dict_m = load_all(find_parameter_files(eclfiles))
assert "FOO" in param_dict
assert "BAR" in param_dict
assert param_dict["BAR"] == "com"
assert param_dict == param_dict_m
os.unlink(parametersjson)
def test_paramsupport(tmpdir):
"""Test that we can merge in parameters.txt
This test code manipulates the paths in the checked out
repository (as it involves some pointing upwards in the directory structure)
It should not leave any extra files around, but requires certain filenames
not to be under version control.
"""
tmpcsvfile = tmpdir / "sum.csv"
eclfiles = EclFiles(DATAFILE)
parameterstxt = Path(eclfiles.get_path()) / "parameters.txt"
if parameterstxt.is_file():
parameterstxt.unlink()
parameterstxt.write_text("FOO 1\nBAR 3", encoding="utf-8")
sys.argv = ["ecl2csv", "summary", DATAFILE, "-o", str(tmpcsvfile), "-p"]
ecl2csv.main()
disk_df = pd.read_csv(tmpcsvfile)
assert "FOPT" in disk_df
assert "FOO" in disk_df
assert "BAR" in disk_df
assert disk_df["BAR"].unique()[0] == 3
parameterstxt.unlink()
parametersyml = Path(eclfiles.get_path()) / "parameters.yml"
if parametersyml.is_file():
parametersyml.unlink()
parametersyml.write_text(yaml.dump({"FOO": 1, "BAR": 3}), encoding="utf-8")
sys.argv = ["ecl2csv", "summary", DATAFILE, "-o", str(tmpcsvfile), "-p"]
ecl2csv.main()
disk_df = pd.read_csv(str(tmpcsvfile))
assert "FOPT" in disk_df
assert "FOO" in disk_df
assert len(disk_df["FOO"].unique()) == 1
assert disk_df["FOO"].unique()[0] == 1
assert "BAR" in disk_df
assert len(disk_df["BAR"].unique()) == 1
assert disk_df["BAR"].unique()[0] == 3
# Test the merging from summary.df() explicitly:
assert "FOO" in summary.df(eclfiles, params=True, paramfile=None)
assert "FOO" not in summary.df(eclfiles, params=False, paramfile=None)
assert "FOO" not in summary.df(eclfiles, params=None, paramfile=None)
assert "FOO" in summary.df(eclfiles, params=False, paramfile=parametersyml)
assert "FOO" in summary.df(eclfiles, params=None, paramfile=parametersyml)
assert "FOO" in summary.df(eclfiles,
params=None,
paramfile="parameters.yml")
# Non-existing relative path is a soft error:
assert "FOO" not in summary.df(eclfiles,
params=None,
paramfile="notexisting/parameters.yml")
# Non-existing absolute path is a hard error:
with pytest.raises(FileNotFoundError):
summary.df(eclfiles,
params=None,
paramfile="/tmp/notexisting/parameters.yml")
parametersyml.unlink()
