def test_write_file():
""" Test write_file functionality """
# create water use totals
wateruse_totals = {
'January': 2.0,
'February': 2.0,
'March': 2.0,
'April': 3.0,
'May': 3.0,
'June': 0.0,
'July': 4.0,
'August': 4.0,
'September': 4.0,
'October': 5.0,
'November': 5.0,
'December': 5.0
}
# write water formatted file
data = fixture["sample_data_dict"]
watertxt.write_file(watertxt_data = data, save_path = os.path.join(os.getcwd(), "tests"))
# write water formatted file with new values set in discharge
data = fixture["sample_data_dict"]
new_discharge_data = np.array([230, 240, 280])
data = watertxt.set_parameter_values(watertxt_data = data, name = "Discharge", values = new_discharge_data)
watertxt.write_file(watertxt_data = data , save_path = os.path.join(os.getcwd(), "tests"), filename = "_WATER-with-new-discharge.txt")
# apply water use
data = fixture["sample_data_dict"]
data = watertxt.apply_wateruse(watertxt_data = data, wateruse_totals = wateruse_totals)
watertxt.write_file(watertxt_data = data , save_path = os.path.join(os.getcwd(), "tests"), filename = "_WATER-with-wateruse.txt")
def test_set_parameter_values():
subsurface_data = np.array([100, 110, 120])
watertxt_data = watertxt.set_parameter_values(watertxt_data = fixture["sample_data_dict"], name = "Subsurface Flow", values = subsurface_data)
expected = {"name": "Subsurface Flow (mm/day)", "index": 1, "data": subsurface_data,
"mean": np.mean(subsurface_data), "max": np.max(subsurface_data), "min": np.min(subsurface_data)}
actual = watertxt.get_parameter(watertxt_data, name = "Subsurface Flow")
nose.tools.assert_equals(actual["name"], expected["name"])
nose.tools.assert_equals(actual["index"], expected["index"])
nose.tools.assert_equals(actual["mean"], expected["mean"])
nose.tools.assert_equals(actual["max"], expected["max"])
nose.tools.assert_equals(actual["min"], expected["min"])
np.testing.assert_equal(actual["data"], expected["data"])
