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)