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_load(tmp_path):
"""Test loading of yml/json/txt files into dictionaries"""
os.chdir(tmp_path)
Path("empty").touch()
assert load("empty") == {} # A warning is logged
# yaml file:
Path("foo.yml").write_text("foo: bar")
assert load("foo.yml") == {"foo": "bar"}
assert load(Path("foo.yml")) == {"foo": "bar"}
# yaml syntax errors:
Path("error.yml").write_text("foo: bar :")
with pytest.raises(ValueError, match="Could not parse error.yml"):
load("error.yml")
# json file:
Path("foo.json").write_text('{\n "foo": "bar"\n}')
assert load("foo.json") == {"foo": "bar"}
# Extension does not matter:
Path("wrongextension.yml").write_text('{\n "foo": "bar"\n}')
assert load("wrongextension.yml") == {"foo": "bar"}
# txt file:
Path("foo.txt").write_text("foo bar")
assert load("foo.txt") == {"foo": "bar"}
# txt file errors:
Path("error.txt").write_text("foo bar com")
with pytest.raises(ValueError, match="Could not parse error.txt"):
load("error.txt") == {"foo": "bar"}
with pytest.raises(FileNotFoundError, match="notexisting not found"):
load("notexisting")
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)