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