def test_simple(self):
event_list_raw_data = [
(2, 4, 40, 50, "trip_5", 1),
]
transit_connections = list(
map(lambda el: Connection(*el), event_list_raw_data))
walk_network = networkx.Graph()
walk_network.add_edge(1, 2, d_walk=20)
walk_network.add_edge(3, 4, d_walk=15)
walk_speed = 1
source_stop = 1
target_stop = 4
transfer_margin = 0
start_time = 0
end_time = 50
pareto_tuples = list()
pareto_tuples.append(
LabelTimeSimple(departure_time=20, arrival_time_target=50))
csa_profile = ConnectionScanProfiler(transit_connections, target_stop,
start_time, end_time,
transfer_margin, walk_network,
walk_speed)
csa_profile.run()
source_stop_profile = csa_profile.stop_profiles[source_stop]
source_stop_pareto_tuples = source_stop_profile.get_final_optimal_labels(
)
self._assert_pareto_labels_equal(pareto_tuples,
source_stop_pareto_tuples)
def test_basics(self):
csa_profile = ConnectionScanProfiler(self.transit_connections,
self.target_stop, self.start_time,
self.end_time,
self.transfer_margin,
self.walk_network,
self.walk_speed)
csa_profile.run()
stop_3_pareto_tuples = csa_profile.stop_profiles[
3].get_final_optimal_labels()
self.assertEqual(len(stop_3_pareto_tuples), 1)
self.assertIn(LabelTimeSimple(32, 35), stop_3_pareto_tuples)
stop_2_pareto_tuples = csa_profile.stop_profiles[
2].get_final_optimal_labels()
self.assertEqual(len(stop_2_pareto_tuples), 2)
self.assertIn(LabelTimeSimple(40, 50), stop_2_pareto_tuples)
self.assertIn(LabelTimeSimple(25, 35), stop_2_pareto_tuples)
source_stop_profile = csa_profile.stop_profiles[self.source_stop]
source_stop_pareto_optimal_tuples = source_stop_profile.get_final_optimal_labels(
)
pareto_tuples = list()
pareto_tuples.append(
LabelTimeSimple(departure_time=10, arrival_time_target=35))
pareto_tuples.append(
LabelTimeSimple(departure_time=20, arrival_time_target=50))
pareto_tuples.append(
LabelTimeSimple(departure_time=32, arrival_time_target=55))
self._assert_pareto_labels_equal(pareto_tuples,
source_stop_pareto_optimal_tuples)
def test_last_leg_is_walk(self):
event_list_raw_data = [(0, 1, 0, 10, "trip_1", 1)]
transit_connections = list(
map(lambda el: Connection(*el), event_list_raw_data))
walk_network = networkx.Graph()
walk_network.add_edge(1, 2, d_walk=20)
walk_speed = 1
source_stop = 0
target_stop = 2
transfer_margin = 0
start_time = 0
end_time = 50
pareto_tuples = list()
pareto_tuples.append(
LabelTimeSimple(departure_time=0, arrival_time_target=30))
csa_profile = ConnectionScanProfiler(transit_connections, target_stop,
start_time, end_time,
transfer_margin, walk_network,
walk_speed)
csa_profile.run()
found_tuples = csa_profile.stop_profiles[
source_stop].get_final_optimal_labels()
self._assert_pareto_labels_equal(found_tuples, pareto_tuples)
def test_target_node_not_in_walk_network(self):
event_list_raw_data = [
(0, 1, 0, 10, "trip_1", 1)
]
transit_connections = list(map(lambda el: Connection(*el), event_list_raw_data))
walk_speed = 2
source_stop = 0
target_stop = 1
transfer_margin = 0
start_time = 0
end_time = 50
walk_network = networkx.Graph()
csa_profile = ConnectionScanProfiler(transit_connections, target_stop,
start_time, end_time, transfer_margin,
walk_network, walk_speed)
csa_profile.run()
source_profile = csa_profile.stop_profiles[source_stop]
self.assertEqual(source_profile.evaluate_earliest_arrival_time_at_target(0, 0), 10)
found_tuples = source_profile.get_final_optimal_labels()
self.assertEqual(len(found_tuples), 1)
def test_wrong_event_data_ordering(self):
event_list_wrong_ordering = [
(0, 1, 0, 10, "trip_1", 1),
(1, 2, 10, 20, "trip_2", 1),
(2, 3, 25, 30, "trip_3", 1),
(3, 4, 32, 35, "trip_4", 1),
(1, 3, 32, 40, "trip_5", 1),
(2, 4, 40, 50, "trip_5", 1)
]
csa_profile = ConnectionScanProfiler(event_list_wrong_ordering, self.target_stop,
self.start_time, self.end_time, self.transfer_margin,
self.walk_network, self.walk_speed)
self.assertRaises(AssertionError, csa_profile.run)
rc("text", usetex=True)
target_stop_I = AALTO_STOP_ID
params = {
"targets": [settings.AALTO_STOP_ID],
"routing_start_time_dep": settings.ROUTING_START_TIME_DEP,
"routing_end_time_dep": settings.ROUTING_END_TIME_DEP
}
connections = compute.read_connections_pandas()
walk_network = compute.read_transfers_csv()
from gtfspy.routing.connection_scan_profile import ConnectionScanProfiler
csa = ConnectionScanProfiler(connections, target_stop_I, settings.ROUTING_START_TIME_DEP,
settings.ANALYSIS_END_TIME_DEP, transfer_margin=180, walk_network=walk_network,
walk_speed=70 / 60.0, verbose=True)
csa.run()
profiles = csa.stop_profiles
from_stop_Is = [
# 123, # Kamppi (as well)
# 401, # Kansanelakelaitos
# 3356, # Dipoli
# 3063, # Kilon asema
# 5935, # Sorvatie
# 3101, # lahderannanristi
# 3373, # Innopoli
# 2843 # Vallikatu (Pohjois-Leppavaara)
ITAKESKUS_STOP_ID,
MUNKKIVUORI_STOP_ID