def culling_jounreys_2_test(): """ max_nb_journeys equals to nb of must-keep journeys ('comfort', 'best') """ mocked_pb_response = build_mocked_response() mocked_request = {'max_nb_journeys': 2, 'debug': False} new_default.culling_journeys(mocked_pb_response, mocked_request) assert len(mocked_pb_response.journeys) == 2 assert all([j.type in new_default.JOURNEY_TYPES_TO_RETAIN for j in mocked_pb_response.journeys])
def culling_jounreys_2_bis_test(): """ max_nb_journeys equals to nb of must-keep journeys ('comfort', 'best', 'non_pt_bike', 'non_pt_walk') Here we test the case where max_nb_journeys equals to nb_journeys_must_have """ mocked_pb_response = build_mocked_response() mocked_request = {'max_nb_journeys': 4, 'debug': False} new_default.culling_journeys(mocked_pb_response, mocked_request) assert len(mocked_pb_response.journeys) == 4 assert all([j.type in new_default.JOURNEY_TYPES_TO_RETAIN for j in mocked_pb_response.journeys])
def culling_jounreys_1_test(): """ Test when max_nb_journeys is bigger than journey's length in response, none of journeys are cut out """ mocked_pb_response = build_mocked_response() nb_journeys = len(mocked_pb_response.journeys) mocked_request = {'max_nb_journeys': nb_journeys + 1, 'debug': False} new_default.culling_journeys(mocked_pb_response, mocked_request) assert len(mocked_pb_response.journeys) == nb_journeys
def culling_jounreys_4_test(): """ When max_nb_journeys == 3 and nb_must_have_journeys == 4 """ mocked_pb_response = build_mocked_response() mocked_request = {'max_nb_journeys': 3, 'debug': False, 'datetime': 1444903200} new_default.culling_journeys(mocked_pb_response, mocked_request) assert len(mocked_pb_response.journeys) == 3 for jrny in mocked_pb_response.journeys: # 'non_pt_bike' shouldn't appear in returned journeys assert jrny.type in ('best', 'comfort', 'non_pt_walk')
def culling_jounreys_3_test(): mocked_pb_response = build_mocked_response() mocked_request = {'max_nb_journeys': 6, 'debug': False, 'datetime': 1444903200} new_default.culling_journeys(mocked_pb_response, mocked_request) assert len(mocked_pb_response.journeys) == 6 journey_uris = { ((u'uri_1', u'uri_2', u'uri_5', u'uri_6', u'walking'), 1444905300), ((u'uri_2', u'uri_3', u'uri_4', u'walking'), 1444905600), ((u'bike', u'uri_2', u'uri_7', u'walking'), 1444907700), ((u'bike', u'uri_9'), 1444905000), ((u'bike', u'uri_8', u'uri_9'), 1444903680), ((u'bike',), 1444903680), ((u'walking',), 1444903500), } for j in mocked_pb_response.journeys: assert (tuple(s.uris.line for s in j.sections), j.arrival_date_time) in journey_uris