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_activity_end_times()
assert plan.day[0].end_time == mtdt(900)
assert plan.day[-1].end_time == END_OF_DAY
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_activity_end_times()
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)
