def test_write_timeseries_file3():
""" Test write_timeseries_file functionality part 3 - Discharge + Water Use; water use is applied """
# 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
}
# create test data
data = fixture["sample_data_dict"]
# apply water use
data = watertxt.apply_wateruse(watertxt_data = data, wateruse_totals = wateruse_totals)
# write file
watertxt.write_timeseries_file(watertxt_data = data, name = "Discharge + Water Use", save_path = os.path.join(os.getcwd(), "tests"), filename = "_q-and-wateruse-timeseries.txt")
def test_write_timeseries_file2():
""" Test write_timeseries_file functionality part 2 - Discharge with only """
# create test data
data = fixture["sample_data_dict"]
# write file
watertxt.write_timeseries_file(watertxt_data = data, name = "Discharge", save_path = os.path.join(os.getcwd(), "tests"), filename = "_q-timeseries.txt")
