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)