def previous_journey_test(): """ In the default scenario, the previous journey is one minute before the first 'rapid' if we get one""" builder = ResponseBuilder(default_date=datetime.date(2016, 10, 10))\ .journey(type='rapid', departure='T1200', arrival='T1500')\ .journey(type='fastest', departure='T1100', arrival='T1700')\ .journey(type='non_pt_walk', departure='T1000', arrival='T1534')\ .journey(type='car', departure='T1300', arrival='T1534') scenario = default.Scenario() eq_(scenario.previous_journey_datetime(builder.get_journeys()), str_to_time_stamp('20161010T145900'))
def previous_journey_test_no_rapid(): """ In the default scenario, if we don't get a rapid, the previous journey is one minute before the tardiest journey """ builder = ResponseBuilder(default_date=datetime.date(2016, 10, 10))\ .journey(type='fastest', departure='T1100', arrival='T1700')\ .journey(type='non_pt_walk', departure='T1000', arrival='T1534')\ .journey(type='car', departure='T2000', arrival='T1534') scenario = default.Scenario() eq_(scenario.previous_journey_datetime(builder.get_journeys()), str_to_time_stamp('20161010T165900'))
def next_journey_test(): """ In the default scenario, the next journey is one minute after the first 'rapid' if we get one""" builder = ResponseBuilder(default_date=datetime.date(2016, 10, 10))\ .journey(type='rapid', departure='T1200', arrival='T1500')\ .journey(type='fastest', departure='T1100', arrival='T1700')\ .journey(type='non_pt_walk', departure='T1000', arrival='T1534')\ .journey(type='car', departure='T1300', arrival='T1534') scenario = default.Scenario() assert scenario.next_journey_datetime( builder.get_journeys(), clockwise=True) == str_to_time_stamp('20161010T120100')