def test_read_file_in():
""" Test read_file_in() """
# description of test
description = "Test read_file_in() - test reading of water use files; using simplified water use formatted file"
# expected values to test with actual values
expected = {"months": "JFM_WU",
"units": "Mgal/day",
"column_names": ["huc12", "newhydroid", "AqGwWL", "CoGwWL", "DoGwWL", "InGwWL", "IrGwWL"],
"huc12": ["20401010101", "20401010101", "20401010101", "20401010101", "20401010101", "20401010101", "20401010102"],
"newhydroid": ["256", "241", "222", "220", "12", "11", "8"],
"AqGwWL": [2.0, 4.0, 6.0, 3.0, 1.0, 2.0, 2.0],
"CoGwWL": [5.0, 3.0, 4.0, 8.0, 3.0, 6.0, 1.0],
"DoGwWL": [2.0, 4.0, 6.0, 3.0, 1.0, 2.0, 2.0],
"InGwWL": [5.0, 3.0, 4.0, 8.0, 3.0, 6.0, 1.0],
"IrGwWL": [-2.0, -4.0, -6.0, -8.0, -1.0, -1.0, -1.0]
}
# create test data
fileobj = StringIO(fixture["data_file_JFM"])
# read file object
actual = wateruse.read_file_in(fileobj)
# assert equality
_perform_assertion(actual, expected, description = description)
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
