def test_find_configurations_for_family_type_0_one_person(self): df = poland.draw_generation_configuration_for_household( self.generations_configuration_df, 1, 0, '', '') self.assertEqual(3, len( df.index)) # only those with nb_generations <= headcount [here: 1]
def test_find_configurations_for_family_type_1_many_people(self): df = poland.draw_generation_configuration_for_household( self.generations_configuration_df, 3, 0, '', '') self.assertEqual(7, len(df.index))
def test_find_configurations_for_family_type_2_related(self): df = poland.draw_generation_configuration_for_household( self.generations_configuration_df, None, 2, 'Spokrewnione w linii prostej', 'członek rodziny starszego pokolenia') self.assertEqual(7, len(df.index))
def test_find_configurations_for_family_type_2_unrelated(self): df = poland.draw_generation_configuration_for_household( self.generations_configuration_df, None, 2, 'Niespokrewnione w linii prostej', '') self.assertEqual(7, len(df.index))
def test_find_configurations_for_family_type_1_selfcontained(self): # df, headcount, family_type, relationship, house_master): df = poland.draw_generation_configuration_for_household( self.generations_configuration_df, None, 1, 'Bez osób spoza rodziny', '') self.assertEqual(7, len(df.index))
def test_find_configurations_for_family_type_1_unrelated(self): # df, headcount, family_type, relationship, house_master): df = poland.draw_generation_configuration_for_household( self.generations_configuration_df, None, 1, 'Z innymi osobami', 'członek rodziny') self.assertEqual(7, len(df.index))
def test_find_configurations_for_family_type_1_related(self): # df, headcount, family_type, relationship, house_master): df = poland.draw_generation_configuration_for_household( self.generations_configuration_df, None, 1, 'Z krewnymi w linii prostej starszego pokolenia', 'inna osoba') self.assertEqual(7, len(df.index))