def build_mocked_response():
response = response_pb2.Response()
for jrny in JOURNEYS:
arrival_time, jrny_type, sections_idx = jrny
pb_j = response.journeys.add()
pb_j.arrival_date_time = arrival_time
# we remove the street network section
pb_j.nb_transfers = len(sections_idx) - 1 - (1 in sections_idx) - (
10 in sections_idx)
if jrny_type:
pb_j.type = jrny_type
for idx in sections_idx:
section_type, network_mode, line_uri = SECTIONS_CHOICES[idx]
section = pb_j.sections.add()
section.type = section_type
if network_mode:
section.street_network.mode = network_mode
if line_uri:
section.uris.line = line_uri
new_default._tag_by_mode([response])
new_default._tag_direct_path([response])
new_default._tag_bike_in_pt([response])
return response
def build_mocked_car_response():
response = response_pb2.Response()
sections = (
(response_pb2.STREET_NETWORK, response_pb2.Walking, 'walking'),
(response_pb2.STREET_NETWORK, response_pb2.Car, 'car'),
(response_pb2.PUBLIC_TRANSPORT, None, 'uri_1'),
(response_pb2.PUBLIC_TRANSPORT, None, 'uri_2'),
)
journeys = (
# J1
# 10 / 15 / 2015 @ 12:30pm (UTC)
# this journey will be tagged 'walking'
(1444905000, 'rapid', (0, 2, 0)),
# J2
# 10 / 15 / 2015 @ 12:30pm (UTC)
# this journey will be tagged 'car'
(1444905000, 'car', (1, 3, 2)),
)
for jrny in journeys:
arrival_time, jrny_type, sections_idx = jrny
pb_j = response.journeys.add()
pb_j.arrival_date_time = arrival_time
# we remove the street network section
pb_j.nb_transfers = len(sections_idx) - 1 - (1 in sections_idx) - (
10 in sections_idx)
if jrny_type:
pb_j.type = jrny_type
for idx in sections_idx:
section_type, network_mode, line_uri = sections[idx]
section = pb_j.sections.add()
section.type = section_type
if network_mode:
section.street_network.mode = network_mode
if line_uri:
section.uris.line = line_uri
new_default._tag_by_mode([response])
new_default._tag_direct_path([response])
new_default._tag_bike_in_pt([response])
return response
def tag_direct_path_test():
response = response_pb2.Response()
# test with one walk and one pt
journey_walk = response.journeys.add()
section = journey_walk.sections.add()
section.type = response_pb2.STREET_NETWORK
section.street_network.mode = response_pb2.Walking
journey_bike = response.journeys.add()
section = journey_bike.sections.add()
section.type = response_pb2.STREET_NETWORK
section.street_network.mode = response_pb2.Bike
journey_pt = response.journeys.add()
section = journey_pt.sections.add()
section.type = response_pb2.PUBLIC_TRANSPORT
new_default._tag_direct_path([response])
assert 'non_pt_walking' in response.journeys[0].tags
assert 'non_pt' in response.journeys[0].tags
assert 'non_pt_bike' in response.journeys[1].tags
assert 'non_pt' in response.journeys[1].tags
assert not response.journeys[2].tags