def test_PersonPolicy_verifies_for_appropriate_probabilities(mocker):
mocker.patch.object(modify, 'verify_probability')
modify.PersonPolicy(modify.RemoveActivity(['']), 0.5)
modify.verify_probability.assert_called_once_with(
0.5,
(float, list, modify.SimpleProbability, modify.ActivityProbability, modify.PersonProbability)
)
def test_PersonPolicy_apply_to_delegates_to_modifier_policy_apply_to_for_list_of_probabilities(mocker, SmithHousehold):
mocker.patch.object(modify.RemoveActivity, 'apply_to')
mocker.patch.object(modify.SimpleProbability, 'p', return_value=1)
policy = modify.PersonPolicy(modify.RemoveActivity(['']), [1., modify.SimpleProbability(1.)])
household = SmithHousehold
policy.apply_to(household)
assert modify.RemoveActivity.apply_to.call_count == 4
def test_person_policy_with_activity_based_probability_with_a_satisfied_person_attribute(
SmithHousehold, mocker):
mocker.patch.object(modify.RemoveActivity, 'remove_person_activities')
mocker.patch.object(random, 'random', side_effect=[0] + [1] * 11)
household = SmithHousehold
# i.e. First of Steve's work activities is affected and affects all listed activities for just Steve
def discrete_sampler(obj, mapping, distribution):
p = distribution
for key in mapping:
value = obj.attributes.get(key)
if value is None:
raise KeyError(
f"Cannot find mapping: {key} in sampling features: {obj.attributes}"
)
p = p.get(value)
if p is None:
raise KeyError(
f"Cannot find feature for {key}: {value} in distribution: {p}"
)
return p
age_mapping = ['age']
below_20 = [i for i in range(21)]
above_20 = [i for i in range(21, 101)]
age_distribution = {
**dict(zip(below_20, [0] * len(below_20))),
**dict(zip(above_20, [1] * len(above_20)))
}
people_satisfying_age_condition_over_20 = 0
for pid, person in household.people.items():
people_satisfying_age_condition_over_20 += discrete_sampler(
person, age_mapping, age_distribution)
assert people_satisfying_age_condition_over_20 == 2
policy = modify.PersonPolicy(
modify.RemoveActivity(
['education', 'escort', 'leisure', 'shop', 'work']), [
modify.ActivityProbability(
['education', 'escort', 'leisure', 'shop', 'work'], 0.5),
modify.PersonProbability(discrete_sampler, {
'mapping': age_mapping,
'distribution': age_distribution
})
])
policy.apply_to(household)
steve = household.people['1']
modify.RemoveActivity.remove_person_activities.assert_called_once_with(
steve)
def test_person_policy_with_person_based_probability(mocker, SmithHousehold):
mocker.patch.object(modify.RemoveActivity, 'remove_person_activities')
mocker.patch.object(random, 'random', side_effect=[1, 1, 1, 0])
household = SmithHousehold
# i.e. Bobby is affected and his activities are the only one affected in household
policy = modify.PersonPolicy(
modify.RemoveActivity(
['education', 'escort', 'leisure', 'shop', 'work']),
modify.PersonProbability(0.5))
bobby = household.people['4']
policy.apply_to(household)
modify.RemoveActivity.remove_person_activities.assert_called_once_with(
bobby)
def test_person_policy_with_activity_based_probability(SmithHousehold, mocker):
mocker.patch.object(modify.RemoveActivity, 'remove_person_activities')
mocker.patch.object(random, 'random', side_effect=[0] + [1] * 11)
household = SmithHousehold
# i.e. First of Steve's work activities is affected and affects all listed activities for just Steve
policy = modify.PersonPolicy(
modify.RemoveActivity(
['education', 'escort', 'leisure', 'shop', 'work']),
modify.ActivityProbability(
['education', 'escort', 'leisure', 'shop', 'work'], 0.5))
policy.apply_to(household)
steve = household.people['1']
modify.RemoveActivity.remove_person_activities.assert_called_once_with(
steve)
def test_person_policy_with_person_based_probability_with_a_satisfied_person_attribute(
mocker, SmithHousehold):
mocker.patch.object(modify.RemoveActivity, 'remove_person_activities')
mocker.patch.object(random, 'random', side_effect=[1, 1, 1, 0])
household = SmithHousehold
# i.e. Bobby is affected and his activities are the only one affected in household
def discrete_sampler(obj, mapping, distribution):
p = distribution
for key in mapping:
value = obj.attributes.get(key)
if value is None:
raise KeyError(
f"Cannot find mapping: {key} in sampling features: {obj.attributes}"
)
p = p.get(value)
if p is None:
raise KeyError(
f"Cannot find feature for {key}: {value} in distribution: {p}"
)
return p
age_mapping = ['age']
below_10 = [i for i in range(11)]
above_10 = [i for i in range(11, 101)]
age_distribution = {
**dict(zip(below_10, [1] * len(below_10))),
**dict(zip(above_10, [0] * len(above_10)))
}
people_satisfying_age_condition_under_10 = 0
for pid, person in household.people.items():
people_satisfying_age_condition_under_10 += discrete_sampler(
person, age_mapping, age_distribution)
assert people_satisfying_age_condition_under_10 == 1
policy = modify.PersonPolicy(
modify.RemoveActivity(
['education', 'escort', 'leisure', 'shop', 'work']),
modify.PersonProbability(discrete_sampler, {
'mapping': age_mapping,
'distribution': age_distribution
}))
bobby = household.people['4']
policy.apply_to(household)
modify.RemoveActivity.remove_person_activities.assert_called_once_with(
bobby)
def test_MoveActivityToHomeLocation_moves_shop_to_home_location():
Hilda = Person(1,
attributes={
'age': 45,
'job': 'influencer',
'gender': 'female'
})
Hilda.add(
Activity(1, 'home', 'a', start_time=mtdt(0), end_time=mtdt(8 * 60)))
Hilda.add(
Leg(1,
'car',
'a',
'b',
start_time=mtdt(8 * 60),
end_time=mtdt(8 * 60 + 30)))
Hilda.add(
Activity(2,
'shop',
'b',
start_time=mtdt(8 * 60 + 30),
end_time=mtdt(16 * 60 + 30)))
Hilda.add(
Leg(1,
'car',
'b',
'a',
start_time=mtdt(16 * 60 + 30),
end_time=mtdt(17 * 60)))
Hilda.add(
Activity(5, 'home', 'a', start_time=mtdt(17 * 60),
end_time=END_OF_DAY))
hhld = instantiate_household_with([Hilda])
policy = modify.PersonPolicy(
modify.MoveActivityTourToHomeLocation(['shop']),
modify.PersonProbability(1))
policy.apply_to(hhld)
assert Hilda.plan[2].location == Hilda.home
assert Hilda.plan[2].is_exact(
Activity(2,
'shop',
'a',
start_time=mtdt(8 * 60 + 30),
end_time=mtdt(16 * 60 + 30)))
def test_MoveActivityToHomeLocation_performs_mode_shift_to_walk_due_to_lack_of_driving_licence(
):
Hilda = Person(1,
attributes={
'age': 45,
'job': 'influencer',
'gender': 'female',
'driving_licence': False
})
Hilda.add(
Activity(1, 'home', 'a', start_time=mtdt(0), end_time=mtdt(8 * 60)))
Hilda.add(
Leg(1,
'pt',
'a',
'b',
start_time=mtdt(8 * 60),
end_time=mtdt(8 * 60 + 30)))
Hilda.add(
Activity(2,
'shop',
'b',
start_time=mtdt(8 * 60 + 30),
end_time=mtdt(16 * 60 + 30)))
Hilda.add(
Leg(1,
'pt',
'b',
'a',
start_time=mtdt(16 * 60 + 30),
end_time=mtdt(17 * 60)))
Hilda.add(
Activity(5, 'home', 'a', start_time=mtdt(17 * 60),
end_time=END_OF_DAY))
hhld = instantiate_household_with([Hilda])
policy = modify.PersonPolicy(
modify.MoveActivityTourToHomeLocation(['shop']),
modify.PersonProbability(1))
policy.apply_to(hhld)
assert Hilda.plan[1].mode == 'walk'
assert Hilda.plan[3].mode == 'walk'
def test_MoveActivityToHomeLocation_does_moves_only_valid_shopping_tour(
SmithHousehold):
household = SmithHousehold
Hilda = Person(2,
attributes={
'age': 45,
'job': 'influencer',
'gender': 'female'
})
Hilda.add(
Activity(1, 'home', 'a', start_time=mtdt(0), end_time=mtdt(8 * 60)))
Hilda.add(
Leg(1,
'walk',
'a',
'b',
start_time=mtdt(8 * 60),
end_time=mtdt(8 * 60 + 30)))
Hilda.add(
Activity(2,
'shop',
'b',
start_time=mtdt(8 * 60 + 30),
end_time=mtdt(10 * 60 + 30)))
Hilda.add(
Leg(2,
'walk',
'a',
'a',
start_time=mtdt(10 * 60 + 30),
end_time=mtdt(11 * 60)))
Hilda.add(
Activity(3,
'home',
'a',
start_time=mtdt(11 * 60),
end_time=mtdt(12 * 60)))
Hilda.add(
Leg(3,
'walk',
'a',
'c',
start_time=mtdt(12 * 60),
end_time=mtdt(12 * 60 + 30)))
Hilda.add(
Activity(4,
'shop',
'c',
start_time=mtdt(12 * 60 + 30),
end_time=mtdt(16 * 60 + 30)))
Hilda.add(
Leg(4,
'walk',
'a',
'c',
start_time=mtdt(16 * 60 + 30),
end_time=mtdt(17 * 60)))
Hilda.add(
Activity(5,
'leisure',
'c',
start_time=mtdt(17 * 60),
end_time=mtdt(17 * 60 + 30)))
Hilda.add(
Leg(5,
'walk',
'c',
'a',
start_time=mtdt(17 * 60 + 30),
end_time=mtdt(18 * 60)))
Hilda.add(
Activity(6, 'home', 'a', start_time=mtdt(18 * 60),
end_time=END_OF_DAY))
household.people['2'] = Hilda
assert_correct_activities(person=Hilda,
ordered_activities_list=[
'home', 'shop', 'home', 'shop', 'leisure',
'home'
])
assert_correct_activities_locations(
person=Hilda,
ordered_activities_locations_list=['a', 'b', 'a', 'c', 'c', 'a'])
policy = modify.PersonPolicy(
modify.MoveActivityTourToHomeLocation(['shop']),
modify.PersonProbability(1))
policy.apply_to(household)
assert_correct_activities(person=Hilda,
ordered_activities_list=[
'home', 'shop', 'home', 'shop', 'leisure',
'home'
])
assert_correct_activities_locations(
person=Hilda,
ordered_activities_locations_list=['a', 'a', 'a', 'c', 'c', 'a'])
def test_MoveActivityToHomeLocation_moves_shopping_tour_to_home_location(
SmithHousehold):
household = SmithHousehold
Steve = household.people['1']
Timmy = household.people['3']
Timmy.plan[4].act = 'shop_1'
Bobby = household.people['4']
Hilda = Person(2,
attributes={
'age': 45,
'job': 'influencer',
'gender': 'female'
})
Hilda.add(
Activity(1, 'home', 'a', start_time=mtdt(0), end_time=mtdt(8 * 60)))
Hilda.add(
Leg(1,
'walk',
'a',
'b',
start_time=mtdt(8 * 60),
end_time=mtdt(8 * 60 + 30)))
Hilda.add(
Activity(2,
'shop_1',
'b',
start_time=mtdt(8 * 60 + 30),
end_time=mtdt(16 * 60 + 30)))
Hilda.add(
Leg(2,
'walk',
'b',
'c',
start_time=mtdt(8 * 60),
end_time=mtdt(8 * 60 + 30)))
Hilda.add(
Activity(3,
'shop_2',
'c',
start_time=mtdt(8 * 60 + 30),
end_time=mtdt(16 * 60 + 30)))
Hilda.add(
Leg(3,
'walk',
'c',
'a',
start_time=mtdt(16 * 60 + 30),
end_time=mtdt(17 * 60)))
Hilda.add(
Activity(4, 'home', 'a', start_time=mtdt(17 * 60),
end_time=END_OF_DAY))
household.people['2'] = Hilda
assert_correct_activities(
person=Steve,
ordered_activities_list=['home', 'work', 'leisure', 'work', 'home'])
assert_correct_activities_locations(
person=Steve,
ordered_activities_locations_list=['a', 'b', 'c', 'b', 'a'])
assert_correct_activities(
person=Hilda,
ordered_activities_list=['home', 'shop_1', 'shop_2', 'home'])
assert_correct_activities_locations(
person=Hilda, ordered_activities_locations_list=['a', 'b', 'c', 'a'])
assert_correct_activities(person=Timmy,
ordered_activities_list=[
'home', 'education', 'shop_1', 'education',
'leisure', 'home'
])
assert_correct_activities_locations(
person=Timmy,
ordered_activities_locations_list=['a', 'b', 'c', 'b', 'd', 'a'])
assert_correct_activities(
person=Bobby, ordered_activities_list=['home', 'education', 'home'])
assert_correct_activities_locations(
person=Bobby, ordered_activities_locations_list=['a', 'b', 'a'])
policy = modify.PersonPolicy(
modify.MoveActivityTourToHomeLocation(['shop_1', 'shop_2']),
modify.PersonProbability(1))
policy.apply_to(household)
assert_correct_activities(
person=Steve,
ordered_activities_list=['home', 'work', 'leisure', 'work', 'home'])
assert_correct_activities_locations(
person=Steve,
ordered_activities_locations_list=['a', 'b', 'c', 'b', 'a'])
assert_correct_activities(
person=Hilda,
ordered_activities_list=['home', 'shop_1', 'shop_2', 'home'])
assert_correct_activities_locations(
person=Hilda, ordered_activities_locations_list=['a', 'a', 'a', 'a'])
assert_correct_activities(person=Timmy,
ordered_activities_list=[
'home', 'education', 'shop_1', 'education',
'leisure', 'home'
])
assert_correct_activities_locations(
person=Timmy,
ordered_activities_locations_list=['a', 'b', 'c', 'b', 'd', 'a'])
assert_correct_activities(
person=Bobby, ordered_activities_list=['home', 'education', 'home'])
assert_correct_activities_locations(
person=Bobby, ordered_activities_locations_list=['a', 'b', 'a'])