def test_population_combine_person_duplicate_key():
pop1 = Population()
hh1 = Household('1')
p1 = Person('1')
hh1.add(p1)
pop1.add(hh1)
p2 = Person('1')
with pytest.raises(KeyError):
pop1.combine(p2, '')
def test_population_combine_person():
pop1 = Population()
hh1 = Household('1')
p1 = Person('1')
hh1.add(p1)
pop1.add(hh1)
p2 = Person('1')
pop1.combine(p2, 'test_')
assert set(pop1.households) == {'1', 'test_1'}
assert set(pop1.households['test_1'].people) == {'test_1'}
assert isinstance(pop1.households['test_1'], Household)
assert isinstance(pop1.households['test_1'].people['test_1'], Person)
