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_convert_wateruse_units():
""" Test convert_wateruse_units() """
description = "Test convert_wateruse_units() which converts from mega gallons per day to cubic feet per second"
# expected values
expected = {"1": 1.5472337962962963,
"5": 7.736168981481482
}
# read file object
actual = {"1" : wateruse.convert_wateruse_units(value = 1.),
"5" : wateruse.convert_wateruse_units(value = 5.)
}
# assert equality
_perform_assertion(actual, expected, description = description)
