def test_person_add_activity_activity_raise_error():
plan = Plan()
act = Activity(1, 'home', 1)
plan.add(act)
act = Activity(2, 'work', 1)
with pytest.raises(PAMSequenceValidationError):
plan.add(act)
def test_activity_tours_segments_home_to_home_looped_plan(activities_and_tour):
plan = Plan(1)
for i in range(len(activities_and_tour['activities']) - 1):
plan.add(activities_and_tour['activities'][i])
plan.add(Leg(1))
plan.add(activities_and_tour['activities'][-1])
assert plan.activity_tours() == activities_and_tour['tours']
def test_matching_activity_tours_delegates_to_plan_activity_tours(
mocker, plain_filter):
mocker.patch.object(Plan, 'activity_tours')
MoveActivityTourToHomeLocation(['']).matching_activity_tours(
Plan(1), plain_filter)
Plan.activity_tours.assert_called_once()
def test_person_add_leg():
plan = Plan()
act = Activity(1, 'home', 1)
plan.add(act)
leg = Leg(1, 'car', start_area=1, end_area=2)
plan.add(leg)
assert len(plan.day) == 2
def test_mode_shift_multiple_tours():
plan = Plan('a')
plan.add(Activity(1, 'home', 'a', start_time=mtdt(0), end_time=mtdt(60)))
plan.add(Leg(1, mode='car', start_time=mtdt(60), end_time=mtdt(90)))
plan.add(Activity(2, 'work', 'b', start_time=mtdt(90), end_time=mtdt(500)))
plan.add(Leg(2, mode='car', start_time=mtdt(500), end_time=mtdt(400)))
plan.add(Activity(3, 'shop', 'b', start_time=mtdt(400),
end_time=mtdt(450)))
plan.add(Leg(3, mode='car', start_time=mtdt(450), end_time=mtdt(600)))
plan.add(Activity(4, 'work', 'b', start_time=mtdt(600),
end_time=mtdt(660)))
plan.add(Leg(4, mode='car', start_time=mtdt(660), end_time=mtdt(800)))
plan.add(Activity(5, 'home', 'a', start_time=mtdt(830),
end_time=mtdt(860)))
plan.add(Leg(5, mode='walk', start_time=mtdt(860), end_time=mtdt(900)))
plan.add(
Activity(6, 'other', 'a', start_time=mtdt(900), end_time=mtdt(920)))
plan.add(Leg(6, mode='walk', start_time=mtdt(920), end_time=mtdt(950)))
plan.add(
Activity(7,
'home',
'a',
start_time=mtdt(950),
end_time=mtdt(24 * 60 - 1)))
plan.mode_shift(5, 'pt')
assert [leg.mode
for leg in plan.legs] == ['pt', 'pt', 'pt', 'pt', 'walk', 'walk']
def test_plan():
plan = Plan()
plan.add(
Activity(seq=1,
act='home',
area='a',
start_time=mtdt(0),
end_time=mtdt(180)))
plan.add(
Leg(seq=1,
mode='car',
start_area='a',
end_area='b',
start_time=mtdt(180),
end_time=mtdt(190)))
plan.add(
Activity(seq=2,
act='work',
area='b',
start_time=mtdt(190),
end_time=mtdt(200)))
plan.add(
Leg(seq=1,
mode='car',
start_area='b',
end_area='a',
start_time=mtdt(200),
end_time=mtdt(390)))
plan.add(
Activity(seq=3,
act='home',
area='b',
start_time=mtdt(390),
end_time=END_OF_DAY))
return plan
def test_person_add_leg_leg_raise_error():
plan = Plan()
act = Activity(1, 'home', 1)
plan.add(act)
leg = Leg(1, 'car', start_area=1, end_area=2)
plan.add(leg)
leg = Leg(2, 'car', start_area=2, end_area=1)
with pytest.raises(PAMSequenceValidationError):
plan.add(leg)
def test_move_activity_at_start_of_plan_updates_leg():
plan = Plan('a')
plan.add(Activity(1, 'shop', 'b'))
plan.add(Leg(1))
plan.add(Activity(2, 'home', 'a'))
plan.move_activity(0)
assert plan[1].start_location == 'a'
def test_move_activity_at_end_of_plan_updates_leg():
plan = Plan('a')
plan.add(Activity(1, 'home', area='a'))
plan.add(Leg(1))
plan.add(Activity(2, 'shop', area='b'))
plan.move_activity(2)
assert plan[1].end_location == 'a'
def test_compare_plans_not_equal():
plana = Plan()
plana.add(
Activity(seq=1,
act='work',
area='A',
start_time=mtdt(0),
end_time=mtdt(600)))
plana.add(
Leg(seq=2,
mode='car',
start_area='A',
end_area='B',
start_time=mtdt(600),
end_time=mtdt(620)))
plana.add(
Activity(seq=3,
act='shop',
area='B',
start_time=mtdt(620),
end_time=mtdt(1200)))
planb = Plan()
planb.add(
Activity(seq=1,
act='work',
area='A',
start_time=mtdt(0),
end_time=mtdt(800)))
planb.add(
Leg(seq=2,
mode='car',
start_area='A',
end_area='B',
start_time=mtdt(800),
end_time=mtdt(620)))
planb.add(
Activity(seq=3,
act='shop',
area='B',
start_time=mtdt(620),
end_time=mtdt(1200)))
assert not plana == planb
def test_reverse_iter():
plan = Plan()
act = Activity(1, 'home', 1, start_time=mtdt(0))
plan.add(act)
leg = Leg(1, 'car', start_area=1, end_area=2, start_time=mtdt(900), end_time=mtdt(930))
plan.add(leg)
act = Activity(2, 'work', 1, start_time=mtdt(930))
plan.add(act)
idxs = list(i for i, c in plan.reversed())
assert idxs == [2,1,0]
def test_leg_duration():
plan = Plan('a')
plan.add(
Activity(seq=1,
act='home',
area='a',
start_time=mtdt(0),
end_time=mtdt(60)))
plan.add(
Leg(seq=1,
mode='car',
start_area='a',
end_area='b',
start_time=mtdt(60),
end_time=mtdt(90)))
plan.add(
Activity(seq=2,
act='work',
area='b',
start_time=mtdt(90),
end_time=mtdt(120)))
plan.add(
Leg(seq=2,
mode='car',
start_area='b',
end_area='a',
start_time=mtdt(120),
end_time=mtdt(180)))
plan.add(
Activity(seq=3,
act='home',
area='a',
start_time=mtdt(180),
end_time=mtdt(24 * 60 - 1)))
plan.mode_shift(3,
'rail',
mode_speed={
'car': 37,
'bus': 10,
'walk': 4,
'cycle': 14,
'pt': 23,
'rail': 37
},
update_duration=True)
assert [act.duration for act in plan] == [
timedelta(seconds=3603),
timedelta(seconds=1800),
timedelta(seconds=1800),
timedelta(seconds=3600),
timedelta(seconds=75597)
]
def test_move_activity_with_home_default():
plan = Plan('a')
plan.add(Activity(1, 'home', area='a'))
plan.add(Leg(1))
plan.add(Activity(2, 'shop', area='b'))
plan.add(Leg(2))
plan.add(Activity(3, 'home', area='a'))
plan.move_activity(2)
assert plan[2].location == 'a'
def test_matching_activity_tours_matches_one_tour(mocker, activity_tours,
plain_filter):
mocker.patch.object(Plan, 'activity_tours', return_value=activity_tours)
mocker.patch.object(MoveActivityTourToHomeLocation,
'tour_matches_activities',
side_effect=[True, False])
matching_tours = MoveActivityTourToHomeLocation(
['']).matching_activity_tours(Plan(1), plain_filter)
assert matching_tours == [activity_tours[0]]
def test_finalise():
plan = Plan()
act = Activity(1, 'home', 1, start_time=mtdt(0))
plan.add(act)
leg = Leg(1, 'car', start_area=1, end_area=2, start_time=mtdt(900), end_time=mtdt(930))
plan.add(leg)
act = Activity(2, 'work', 1, start_time=mtdt(930))
plan.add(act)
plan.finalise()
assert plan.day[0].end_time == mtdt(900)
assert plan.day[-1].end_time == END_OF_DAY
def test_move_activity_with_different_default():
plan = Plan('a')
plan.add(Activity(1, 'home', area='a'))
plan.add(Leg(1))
plan.add(Activity(2, 'shop', area='b'))
plan.add(Leg(2))
plan.add(Activity(3, 'home', area='a'))
new_loc = Location(area='heyooo')
plan.move_activity(2, default=new_loc)
assert plan[2].location == new_loc
def test_move_activity_with_home_default_updates_legs():
plan = Plan('a')
plan.add(Activity(1, 'home', area='a'))
plan.add(Leg(1))
plan.add(Activity(2, 'shop', area='b'))
plan.add(Leg(2))
plan.add(Activity(3, 'home', area='a'))
plan.move_activity(2)
assert plan[1].end_location == 'a'
assert plan[3].start_location == 'a'
def test_matching_activity_tours_delegates_to_tour_matches_activities(
mocker, activity_tours, plain_filter):
mocker.patch.object(Plan,
'activity_tours',
return_value=[activity_tours[0]])
mocker.patch.object(MoveActivityTourToHomeLocation,
'tour_matches_activities')
MoveActivityTourToHomeLocation(['']).matching_activity_tours(
Plan(1), plain_filter)
MoveActivityTourToHomeLocation.tour_matches_activities.assert_called_once_with(
activity_tours[0], plain_filter)
def test_duration():
plan = Plan()
act = Activity(1, 'home', 1, start_time=mtdt(0))
plan.add(act)
leg = Leg(1, 'car', start_area=1, end_area=2, start_time=mtdt(900), end_time=mtdt(930))
plan.add(leg)
act = Activity(2, 'work', 1, start_time=mtdt(930))
plan.add(act)
plan.finalise()
assert plan.day[0].duration == timedelta(minutes=900)
assert plan.day[1].duration == timedelta(minutes=30)
assert plan.day[-1].duration == timedelta(seconds=(24*60-930)*60)
def test_activity_tours_segments_home_to_other_act_nonlooped_plan(
activities_and_tour):
plan = Plan(1)
for i in range(len(activities_and_tour['activities'])):
plan.add(activities_and_tour['activities'][i])
plan.add(Leg(1))
other_act = Activity(8, 'other', 'e')
plan.add(other_act)
assert plan[0].act != plan[-1].act
assert plan.activity_tours(
) == activities_and_tour['tours'] + [[other_act]]
def test_move_activity_with_different_default_updates_legs():
plan = Plan('a')
plan.add(Activity(1, 'home', area='a'))
plan.add(Leg(1))
plan.add(Activity(2, 'shop', area='b'))
plan.add(Leg(2))
plan.add(Activity(3, 'home', area='a'))
new_loc = Location(area='heyooo')
plan.move_activity(2, default=new_loc)
assert plan[1].end_location == new_loc
assert plan[3].start_location == new_loc
def test_mode_shift_single_tour():
plan = Plan('a')
plan.add(Activity(1, 'home', 'a', start_time=mtdt(0), end_time=mtdt(60)))
plan.add(Leg(1, mode='car', start_time=mtdt(60), end_time=mtdt(90)))
plan.add(Activity(2, 'shop', 'b', start_time=mtdt(90), end_time=mtdt(500)))
plan.add(Leg(2, mode='car', start_time=mtdt(500), end_time=mtdt(400)))
plan.add(
Activity(3,
'home',
'a',
start_time=mtdt(400),
end_time=mtdt(24 * 60 - 1)))
plan.mode_shift(1, 'pt')
assert [leg.mode for leg in plan.legs] == ['pt', 'pt']
def test_plan_fix():
plan = Plan()
plan.add(
Activity(seq=1,
act='home',
area='A',
start_time=mtdt(0),
end_time=mtdt(600)))
plan.add(
Leg(seq=2,
mode='car',
start_area='B',
end_area='B',
start_time=mtdt(610),
end_time=mtdt(620)))
plan.add(
Activity(seq=3,
act='work',
area='B',
start_time=mtdt(620),
end_time=mtdt(12000)))
plan.add(
Leg(seq=2,
mode='car',
start_area='B',
end_area='A',
start_time=mtdt(12000),
end_time=mtdt(11000)))
plan.add(
Activity(seq=1,
act='home',
area='A',
start_time=mtdt(11000),
end_time=END_OF_DAY))
plan.fix()
assert plan.length == 3
assert plan.day[-1].end_time == END_OF_DAY
assert plan[1].start_time == mtdt(600)
assert plan[1].start_location.area == 'A'
def test_return_first_act_as_home_act_if_missing():
plan = Plan()
plan.add(
Activity(seq=1,
act='work',
area='A',
start_time=mtdt(0),
end_time=mtdt(600)))
plan.add(
Leg(seq=2,
mode='car',
start_area='A',
end_area='B',
start_time=mtdt(600),
end_time=mtdt(620)))
plan.add(
Activity(seq=3,
act='shop',
area='B',
start_time=mtdt(620),
end_time=mtdt(1200)))
assert plan.home == 'A'
def test_mode_shift_two_tours_third_leg():
plan = Plan('a')
plan.add(Activity(1, 'home', 'a', start_time=mtdt(0), end_time=mtdt(60)))
plan.add(Leg(1, mode='car', start_time=mtdt(60), end_time=mtdt(90)))
plan.add(Activity(2, 'shop', 'b', start_time=mtdt(90), end_time=mtdt(500)))
plan.add(Leg(2, mode='car', start_time=mtdt(500), end_time=mtdt(400)))
plan.add(Activity(3, 'home', 'a', start_time=mtdt(400),
end_time=mtdt(860)))
plan.add(Leg(3, mode='car', start_time=mtdt(860), end_time=mtdt(900)))
plan.add(Activity(4, 'work', 'a', start_time=mtdt(920),
end_time=mtdt(930)))
plan.add(Leg(4, mode='car', start_time=mtdt(930), end_time=mtdt(1000)))
plan.add(
Activity(5,
'home',
'a',
start_time=mtdt(1000),
end_time=mtdt(24 * 60 - 1)))
plan.mode_shift(5, 'pt')
assert [leg.mode for leg in plan.legs] == ['car', 'car', 'pt', 'pt']
def test_fix_time_consistency():
plan = Plan()
plan.add(
Activity(seq=1,
act='home',
area='A',
start_time=mtdt(0),
end_time=mtdt(600)))
plan.add(
Leg(seq=2,
mode='car',
start_area='A',
end_area='B',
start_time=mtdt(610),
end_time=mtdt(620)))
plan.add(
Activity(seq=3,
act='work',
area='B',
start_time=mtdt(620),
end_time=END_OF_DAY))
plan.fix_time_consistency()
assert plan[1].start_time == mtdt(600)
def test_crop_act_out_of_order():
plan = Plan()
plan.add(
Activity(seq=1,
act='home',
area='A',
start_time=mtdt(0),
end_time=mtdt(600)))
plan.add(
Leg(seq=2,
mode='car',
start_area='A',
end_area='B',
start_time=mtdt(600),
end_time=mtdt(620)))
plan.add(
Activity(seq=3,
act='work',
area='B',
start_time=mtdt(620),
end_time=mtdt(12000)))
plan.add(
Leg(seq=2,
mode='car',
start_area='B',
end_area='A',
start_time=mtdt(12000),
end_time=END_OF_DAY))
plan.add(
Activity(seq=1,
act='home',
area='A',
start_time=mtdt(0),
end_time=mtdt(12030)))
plan.crop()
assert plan.length == 3
assert plan.day[-1].end_time == END_OF_DAY
def test_compare_plans_not_equal_types():
plana = Plan()
plana.add(
Activity(seq=1,
act='work',
area='A',
start_time=mtdt(0),
end_time=mtdt(600)))
plana.add(
Leg(seq=2,
mode='car',
start_area='A',
end_area='B',
start_time=mtdt(600),
end_time=mtdt(620)))
plana.add(
Activity(seq=3,
act='shop',
area='B',
start_time=mtdt(620),
end_time=mtdt(1200)))
with pytest.raises(UserWarning):
assert not plana == None
def test_get_component():
plan = Plan()
plan.add(
Activity(seq=1,
act='home',
area='A',
start_time=mtdt(0),
end_time=mtdt(600)))
plan.add(
Leg(seq=2,
mode='car',
start_area='A',
end_area='B',
start_time=mtdt(600),
end_time=mtdt(620)))
plan.add(
Activity(seq=3,
act='work',
area='B',
start_time=mtdt(620),
end_time=mtdt(1200)))
assert plan.get(0).act == 'home'
assert plan.get(4) is None
assert plan.get(4, '1') == '1'
def test_mode_and_activity_classes():
plan = Plan()
plan.add(
Activity(seq=1,
act='home',
area='A',
start_time=mtdt(0),
end_time=mtdt(600)))
plan.add(
Leg(seq=2,
mode='car',
start_area='A',
end_area='B',
start_time=mtdt(600),
end_time=mtdt(620)))
plan.add(
Activity(seq=3,
act='work',
area='B',
start_time=mtdt(620),
end_time=mtdt(1200)))
assert plan.activity_classes == set(['home', 'work'])
assert plan.mode_classes == set(['car'])