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"])