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)