def test_stea(mock_stea):
mock_stea.side_effect = calculate_patch
main_entry_point(["-c", "stea_input.yml"])
mock_stea.assert_called_once()
files = os.listdir(os.getcwd())
# the resulting file i.e. key is defined in the input config file: stea_input.yml
assert "NPV_0" in files
def test_stea(mock_calculate, monkeypatch):
monkeypatch.setattr(sys, "argv", ["script_name", "-c", "stea_input.yml"])
fm_stea.main_entry_point()
mock_calculate.assert_called_once()
files = os.listdir(os.getcwd())
# the resulting file i.e. key is defined in the input config file: stea_input.yml
assert "NPV_0" in files
assert "stea_response.json" in files
def test_stea_response(monkeypatch):
expected_result = {
"response": [{"TaxMode": "Corporate", "Values": {"NPV": 30}}],
"profiles": {"a_very_long_string": {"Id": "a_very_long_string"}},
}
monkeypatch.setattr(sys, "argv", ["script_name", "-c", "stea_input.yml"])
fm_stea.main_entry_point()
with open("stea_response.json", "r") as fin:
result = json.load(fin)
assert result == expected_result
def test_stea_ecl_case_overwrite(monkeypatch):
"""
We want to verify that the ecl_case argument is properly forwarded to stea, but
there is no ecl_case, so we just assert that we fail with our custom file missing
"""
monkeypatch.setattr(
sys,
"argv",
["script_name", "-c", "stea_input.yml", "--ecl_case", "custom_ecl_case"],
)
with pytest.raises(
OSError, match="Failed to create summary instance from argument:custom_ecl_case"
):
fm_stea.main_entry_point()
def test_stea_default_ert_args(monkeypatch):
"""
We want to verify that the default ert args are correct, so basically that the
script does not fail.
"""
monkeypatch.setattr(
sys,
"argv",
[
"script_name",
"-c",
"stea_input.yml",
"-e",
"__NONE__",
"-r",
"stea_response.json",
],
)
fm_stea.main_entry_point()
