def test_assign_same_locs_to_household_activity_in_same_area(SmithHousehold): population = Population() population.add(SmithHousehold) class FakeSampler: def sample(self, location_idx, activity): return random() population.sample_locs(FakeSampler()) SmithHousehold['3'].plan[2].location == SmithHousehold['4'].plan[ 2].location
def test_population_sample_locs(person_heh): population = Population() population.add(Household('1')) population['1'].add(person_heh) class DummySampler: def sample(self, loc, act): return None population.sample_locs(DummySampler()) assert population['1']['1'].plan[2].location.loc is None
def test_assign_same_locs_to_household(SmithHousehold): population = Population() population.add(SmithHousehold) class FakeSampler: def sample(self, location_idx, activity): return random() population.sample_locs(FakeSampler()) home_location = population['1'].location for pid, person in SmithHousehold: assert person.home == home_location