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_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]]