def _get_all_total_wateruse_for_tests(wateruse_files, id_list, wateruse_factor_file = None, in_cfs = False):
""" Test get_all_total_wateruse - strictly a test function here that mirrors get_all_total_wateruse in water.py but creates fileobj from StringIO instead of reading a file path """
# calculate average values for a list of water use files
all_total_wateruse_dict = {}
for wateruse_file in wateruse_files:
fileobj = StringIO(wateruse_file)
wateruse_data = wateruse.read_file_in(fileobj)
# if water use factor file is supplied, then apply factors
if wateruse_factor_file:
# read water use factor file
fileobj_factors = StringIO(wateruse_factor_file)
wateruse_factors = wateruse.read_factor_file_in(fileobj_factors)
# calculate average wateruse for a list of ids
total_wateruse_dict = wateruse.get_total_wateruse(wateruse_data = wateruse_data, id_list = id_list, wateruse_factors = wateruse_factors)
else:
# calculate average wateruse for a list of ids
total_wateruse_dict = wateruse.get_total_wateruse(wateruse_data = wateruse_data, id_list = id_list)
# convert values to cfs
if in_cfs:
for key, value in total_wateruse_dict.iteritems():
value_cfs = wateruse.convert_wateruse_units(value)
total_wateruse_dict[key] = value_cfs
# update dictionary
all_total_wateruse_dict.update(total_wateruse_dict)
return all_total_wateruse_dict
def test_get_total_wateruse5():
""" Test get_total_wateruse() part 5 - ids [12, 11, 8] WITH water use factors"""
# description of test
description = "Test get_total_wateruse() : part 5 - test getting the total sum of water use for ids (hydroids) [256, 241, 222, 220] WITH water use factors that are VARIABLE"
# expected values to test with actual values
expected = {"January": 92.0,
"February": 92.0,
"March": 92.0,
}
# actual values
actual = wateruse.get_total_wateruse(wateruse_data = fixture["wateruse_data"], id_list = ["12", "11", "8"], wateruse_factors = fixture["wateruse_factors_variable"])
# assert equality
_perform_assertion(actual, expected, description = description)
def test_get_total_wateruse3():
""" Test get_total_wateruse() part 3 - ids [256, 241, 222, 220] WITH water use factors"""
# description of test
description = "Test get_total_wateruse() : part 3 - test getting the total sum of water use for ids (hydroids) [256, 241, 222, 220] WITH water use factors"
# expected values to test with actual values
expected = {"January": 100.0,
"February": 100.0,
"March": 100.0,
}
# actual values
actual = wateruse.get_total_wateruse(wateruse_data = fixture["wateruse_data"], id_list = ["256", "241", "222", "220"], wateruse_factors = fixture["wateruse_factors"])
# assert equality
_perform_assertion(actual, expected, description = description)
def test_get_total_wateruse2():
""" Test get_total_wateruse() part 2 - ids [12, 11, 8] """
# description of test
description = "Test get_total_wateruse() : part 2 - test getting the total sum of water use for ids (hydroids) [12, 11, 8]"
# expected values to test with actual values
expected = {"January": 27.0,
"February": 27.0,
"March": 27.0,
}
# actual values
actual = wateruse.get_total_wateruse(wateruse_data = fixture["wateruse_data"], id_list = ["12", "11", "8"])
# assert equality
_perform_assertion(actual, expected, description = description)
