def test_csv_loads_same_as_dat(self):
headage_householdsize_distribution_path = spdd.get_household_head_age_by_size_path(
datadir=sp_datadir,
state_location="Washington",
country_location="usa")
headage_distro_datfilename, headage_distro_csvfilename = self.copy_dat_and_make_csv(
headage_householdsize_distribution_path)
headage_distro_dat_df = spdd.get_household_head_age_by_size_df(
datadir="", file_path=headage_distro_datfilename)
headage_distro_csv_df = spdd.get_household_head_age_by_size_df(
datadir="", file_path=headage_distro_csvfilename)
if self.is_debugging:
print(headage_distro_dat_df.describe())
print(headage_distro_dat_df.columns)
dat_household_age_18_20_weights = headage_distro_dat_df[
'household_head_age_18_20']
dat_household_size_4_reference_35_39 = headage_distro_dat_df[
'household_head_age_35_39'][5]
dat_household_size_1_weights = headage_distro_dat_df.loc[0]
if self.is_debugging:
print("Household age 18_20 weights")
print(dat_household_age_18_20_weights)
print("Household size 1 relative weights")
print(dat_household_size_1_weights)
print("Household size 4, reference person 25-39 relative weight")
print(dat_household_size_4_reference_35_39)
csv_household_age_18_20_weights = headage_distro_csv_df[
'household_head_age_18_20']
csv_household_size_4_reference_35_39 = headage_distro_csv_df[
'household_head_age_35_39'][5]
self.assertTrue(
csv_household_age_18_20_weights.equals(
csv_household_age_18_20_weights),
msg="household_head_age_18_20 columns should be the same")
self.assertEqual(csv_household_size_4_reference_35_39,
dat_household_size_4_reference_35_39,
msg="Each cell in the data table should be identical")
def test_get_head_age_by_size_path():
hha_by_size = spdd.get_household_head_age_by_size_path(
datadir,
state_location=state_location,
country_location=country_location)
assert hha_by_size is not None