def test_feed_to_graph_performance():
# Replicate the original workflow of the graph creation path
# but open up to expose to benchmarking/performance profiling
start = 7 * 60 * 60
end = 10 * 60 * 60
interpolate_times = True
use_multiprocessing = False
print('Running time profiles on each major '
'function in graph generation workflow')
a = time()
path = fixture('samtrans-2017-11-28.zip')
feed = get_representative_feed(path)
elapsed = round(time() - a, 2)
print('Perf of get_representative_feed: {}s'.format(elapsed))
fl = len(feed.routes)
print('Iteration on {} routes.'.format(fl))
a = time()
(all_edge_costs,
all_wait_times) = generate_edge_and_wait_values(feed, start, end,
interpolate_times,
use_multiprocessing)
elapsed = round(time() - a, 2)
print('Perf of generate_edge_and_wait_values: {}s'.format(elapsed))
a = time()
summary_edge_costs = generate_summary_edge_costs(all_edge_costs)
elapsed = round(time() - a, 2)
print('Perf of generate_summary_edge_costs: {}s'.format(elapsed))
a = time()
wait_times_by_stop = generate_summary_wait_times(
all_wait_times, FALLBACK_STOP_COST_DEFAULT)
elapsed = round(time() - a, 2)
print('Perf of generate_summary_wait_times: {}s'.format(elapsed))
a = time()
G = generate_empty_md_graph('foo')
elapsed = round(time() - a, 2)
print('Perf of generate_empty_md_graph: {}s'.format(elapsed))
a = time()
G = populate_graph(G, 'bar', feed, wait_times_by_stop, summary_edge_costs,
50, 4.5)
elapsed = round(time() - a, 2)
print('Perf of populate_graph: {}s'.format(elapsed))
def test_generate_empty_graph():
G = generate_empty_md_graph('foo')
assert len(G.edges()) == 0
assert len(G.nodes()) == 0