def test_export_nodes():
with test_db_session() as session:
g = igraph.Graph(directed=True)
# two boxes
g.add_vertices(6)
g.vs["osm_id"] = range(6)
# the outer rectange
g.add_edges([
(1, 0),
(0, 5),
(2, 1),
(2, 3),
(4, 3),
(4, 5),
])
g.es["weight"] = [x + 1 for x in range(6)]
g.vs["redundant"] = [False for _ in range(6)]
for i in range(6):
new_node = OSRMNode(i, 3400000, 11800000, False, False)
session.add(new_node)
session.commit()
export_nodes(g, session)
results = session.query(OSRMRouteNode).filter_by(osm_id=4).all()
eq_(gt.output_weights(g, 4), [5, 6])
eq_(results[0].n_outputs, 2)
eq_(results[0].sum_out, 5 + 6)
eq_(results[0].product_out, 5 * 6)
assert_almost_equal(results[0].log_sum_out, math.log(5) + math.log(6))
def test_insert_node():
with test_db_session() as session:
new_node = OSRMNode(1, 3400000, 11800000, False, False)
session.add(new_node)
session.commit()
# check if we can query it
results = session.query(OSRMNode).filter_by(osm_id=1).all()
eq_(len(results), 1)
eq_(results[0].lat, 3400000)
eq_(results[0].lon, 11800000)
eq_(results[0].bollard, False)
eq_(results[0].traffic_light, False)
eq_(len(results[0].in_edges), 0)
eq_(len(results[0].out_edges), 0)
# check if we can spatial query it
covering_box = WKTSpatialElement(
'POLYGON((117 33, 119 33, 119 35, 117 35, 117 33))')
eq_(
session.query(OSRMNode).filter(
OSRMNode.geom.covered_by(covering_box)).count(), 1)
noncovering_box = WKTSpatialElement(
'POLYGON((117 38, 119 38, 119 39, 117 39, 117 38))')
eq_(
session.query(OSRMNode).filter(
OSRMNode.geom.covered_by(noncovering_box)).count(), 0)
def test_export_nodes():
with test_db_session() as session:
g = igraph.Graph(directed=True)
# two boxes
g.add_vertices(6)
g.vs["osm_id"] = range(6)
# the outer rectange
g.add_edges([
(1, 0),
(0, 5),
(2, 1),
(2, 3),
(4, 3),
(4, 5),
])
g.es["weight"] = [x + 1 for x in range(6)]
g.vs["redundant"] = [False for _ in range(6)]
for i in range(6):
new_node = OSRMNode(i, 3400000, 11800000, False, False)
session.add(new_node)
session.commit()
export_nodes(g, session)
results = session.query(OSRMRouteNode).filter_by(osm_id=4).all()
eq_(gt.output_weights(g, 4), [5, 6])
eq_(results[0].n_outputs, 2)
eq_(results[0].sum_out, 5 + 6)
eq_(results[0].product_out, 5 * 6)
assert_almost_equal(results[0].log_sum_out,
math.log(5) + math.log(6))
def test_freq_backref():
with test_db_session() as session:
node1 = OSRMNode(1, 1, 2, True, False)
node2 = OSRMNode(2, 3, 4, False, False)
session.add(node1)
session.add(node2)
the_edge = OSRMEdge(1, 2, 5, 5, False)
session.add(the_edge)
session.commit()
the_freq = OSRMEdgeFrequencies(
edge=OSRMEdge.hash_edge(1, 2),
forward=True,
freq=10,
geom=None
)
session.add(the_freq)
the_freq_backward = OSRMEdgeFrequencies(
edge=OSRMEdge.hash_edge(1, 2),
forward=False,
freq=11,
geom=None
)
session.add(the_freq_backward)
session.commit()
results = session.query(OSRMEdgeFrequencies).all()
eq_(len(results), 2)
eq_(results[0].freq, 10)
eq_(results[0].forward, True)
# test join
eq_(results[0].edgeobj.source, 1)
eq_(results[0].edgeobj.sink, 2)
def test_freq_backref():
with test_db_session() as session:
node1 = OSRMNode(1, 1, 2, True, False)
node2 = OSRMNode(2, 3, 4, False, False)
session.add(node1)
session.add(node2)
the_edge = OSRMEdge(1, 2, 5, 5, False)
session.add(the_edge)
session.commit()
the_freq = OSRMEdgeFrequencies(edge=OSRMEdge.hash_edge(1, 2),
forward=True,
freq=10,
geom=None)
session.add(the_freq)
the_freq_backward = OSRMEdgeFrequencies(edge=OSRMEdge.hash_edge(1, 2),
forward=False,
freq=11,
geom=None)
session.add(the_freq_backward)
session.commit()
results = session.query(OSRMEdgeFrequencies).all()
eq_(len(results), 2)
eq_(results[0].freq, 10)
eq_(results[0].forward, True)
# test join
eq_(results[0].edgeobj.source, 1)
eq_(results[0].edgeobj.sink, 2)
def test_insert_osrm_routestep():
with test_db_session() as session:
route_hash = OSRMRoute.hash_route(1)
route = OSRMRoute(route_hash=route_hash,
start_lat=2,
start_lon=3,
end_lat=4,
end_lon=5,
duration=6,
nsteps=7)
session.add(route)
node1 = OSRMNode(1, 1, 2, True, False)
node2 = OSRMNode(2, 3, 4, False, False)
node3 = OSRMNode(3, 5, 6, False, False)
session.add(node1)
session.add(node2)
session.add(node3)
edge1 = OSRMEdge(1, 2, 5, 0.3, True)
edge2 = OSRMEdge(2, 3, 6, 0.3, True)
session.add(edge1)
session.add(edge2)
step1 = OSRMRouteStep(route_hash=route_hash,
step_idx=0,
edge_id=OSRMEdge.hash_edge(1, 2),
forward=OSRMEdge.is_forward(1, 2))
step2 = OSRMRouteStep(route_hash=route_hash,
step_idx=1,
edge_id=OSRMEdge.hash_edge(3, 2),
forward=OSRMEdge.is_forward(3, 2))
session.add(step1)
session.add(step2)
session.commit()
result = session.query(OSRMRoute).filter(
OSRMRoute.route_hash == route_hash).all()
eq_(len(result), 1)
eq_(result[0].start_lat, 2)
eq_(result[0].nsteps, 7)
steps = session.query(OSRMRouteStep).all()
eq_(steps[0].edge_id, OSRMEdge.hash_edge(1, 2))
eq_(steps[0].edge_id, OSRMEdge.hash_edge(2, 1)) # commutes
eq_(steps[0].step_idx, 0)
eq_(steps[0].forward, True)
eq_(steps[1].step_idx, 1)
eq_(steps[1].forward, False)
eq_(steps[0].edge.source_node.lon, 2)
eq_(steps[0].edge.source_node.lat, 1)
def test_insert_osrm_route():
with test_db_session() as session:
route_hash = OSRMRoute.hash_route(1)
route = OSRMRoute(route_hash=route_hash, start_lat=2, start_lon=3,
end_lat=4, end_lon=5, duration=6, nsteps=7,
query='a long query string that you can set')
session.add(route)
session.commit()
result = session.query(OSRMRoute).filter(
OSRMRoute.route_hash == OSRMRoute.hash_route(2)).all()
eq_(len(result), 0)
result = session.query(OSRMRoute).filter(
OSRMRoute.route_hash == OSRMRoute.hash_route(1)).all()
eq_(len(result), 1)
eq_(result[0].start_lat, 2)
eq_(result[0].nsteps, 7)
eq_(result[0].query, 'a long query string that you can set')
def test_insert_osrm_routestep():
with test_db_session() as session:
route_hash = OSRMRoute.hash_route(1)
route = OSRMRoute(route_hash=route_hash, start_lat=2, start_lon=3,
end_lat=4, end_lon=5, duration=6, nsteps=7)
session.add(route)
node1 = OSRMNode(1, 1, 2, True, False)
node2 = OSRMNode(2, 3, 4, False, False)
node3 = OSRMNode(3, 5, 6, False, False)
session.add(node1)
session.add(node2)
session.add(node3)
edge1 = OSRMEdge(1, 2, 5, 0.3, True)
edge2 = OSRMEdge(2, 3, 6, 0.3, True)
session.add(edge1)
session.add(edge2)
step1 = OSRMRouteStep(route_hash=route_hash, step_idx=0,
edge_id=OSRMEdge.hash_edge(1, 2),
forward=OSRMEdge.is_forward(1, 2))
step2 = OSRMRouteStep(route_hash=route_hash, step_idx=1,
edge_id=OSRMEdge.hash_edge(3, 2),
forward=OSRMEdge.is_forward(3, 2))
session.add(step1)
session.add(step2)
session.commit()
result = session.query(OSRMRoute).filter(
OSRMRoute.route_hash == route_hash).all()
eq_(len(result), 1)
eq_(result[0].start_lat, 2)
eq_(result[0].nsteps, 7)
steps = session.query(OSRMRouteStep).all()
eq_(steps[0].edge_id, OSRMEdge.hash_edge(1, 2))
eq_(steps[0].edge_id, OSRMEdge.hash_edge(2, 1)) # commutes
eq_(steps[0].step_idx, 0)
eq_(steps[0].forward, True)
eq_(steps[1].step_idx, 1)
eq_(steps[1].forward, False)
eq_(steps[0].edge.source_node.lon, 2)
eq_(steps[0].edge.source_node.lat, 1)
def test_insert_edge():
with test_db_session() as session:
node1 = OSRMNode(1, 1, 2, True, False)
node2 = OSRMNode(2, 3, 4, False, False)
session.add(node1)
session.add(node2)
the_edge = OSRMEdge(1, 2, 5, 5, False)
session.add(the_edge)
session.commit()
result = session.query(OSRMEdge).join(OSRMEdge.source_node).\
filter(OSRMNode.lat == 1).all()
eq_(len(result), 1)
eq_(result[0].source_node.lon, 2)
eq_(result[0].source_node.bollard, True)
eq_(result[0].source_node.out_edges[0].distance, 5)
eq_(len(result[0].source_node.in_edges), 0)
eq_(len(result[0].sink_node.in_edges), 1)
eq_(len(result[0].source_node.out_edges), 1)
eq_(len(result[0].sink_node.out_edges), 0)
def test_insert_edge():
with test_db_session() as session:
node1 = OSRMNode(1, 1, 2, True, False)
node2 = OSRMNode(2, 3, 4, False, False)
session.add(node1)
session.add(node2)
the_edge = OSRMEdge(1, 2, 5, 5, False)
session.add(the_edge)
session.commit()
result = session.query(OSRMEdge).join(OSRMEdge.source_node).\
filter(OSRMNode.lat == 1).all()
eq_(len(result), 1)
eq_(result[0].source_node.lon, 2)
eq_(result[0].source_node.bollard, True)
eq_(result[0].source_node.out_edges[0].distance, 5)
eq_(len(result[0].source_node.in_edges), 0)
eq_(len(result[0].sink_node.in_edges), 1)
eq_(len(result[0].source_node.out_edges), 1)
eq_(len(result[0].sink_node.out_edges), 0)
def test_insert_osrm_route():
with test_db_session() as session:
route_hash = OSRMRoute.hash_route(1)
route = OSRMRoute(route_hash=route_hash,
start_lat=2,
start_lon=3,
end_lat=4,
end_lon=5,
duration=6,
nsteps=7,
query='a long query string that you can set')
session.add(route)
session.commit()
result = session.query(OSRMRoute).filter(
OSRMRoute.route_hash == OSRMRoute.hash_route(2)).all()
eq_(len(result), 0)
result = session.query(OSRMRoute).filter(
OSRMRoute.route_hash == OSRMRoute.hash_route(1)).all()
eq_(len(result), 1)
eq_(result[0].start_lat, 2)
eq_(result[0].nsteps, 7)
eq_(result[0].query, 'a long query string that you can set')
def test_insert_node():
with test_db_session() as session:
new_node = OSRMNode(1, 3400000, 11800000, False, False)
session.add(new_node)
session.commit()
# check if we can query it
results = session.query(OSRMNode).filter_by(osm_id=1).all()
eq_(len(results), 1)
eq_(results[0].lat, 3400000)
eq_(results[0].lon, 11800000)
eq_(results[0].bollard, False)
eq_(results[0].traffic_light, False)
eq_(len(results[0].in_edges), 0)
eq_(len(results[0].out_edges), 0)
# check if we can spatial query it
covering_box = WKTSpatialElement(
'POLYGON((117 33, 119 33, 119 35, 117 35, 117 33))')
eq_(session.query(OSRMNode).filter(
OSRMNode.geom.covered_by(covering_box)).count(), 1)
noncovering_box = WKTSpatialElement(
'POLYGON((117 38, 119 38, 119 39, 117 39, 117 38))')
eq_(session.query(OSRMNode).filter(
OSRMNode.geom.covered_by(noncovering_box)).count(), 0)
def test_build_graph():
with test_db_session() as session:
# Make a y-shaped graph
node1 = OSRMNode(1, 1, 2, True, False)
node2 = OSRMNode(2, 3, 4, False, False)
node3 = OSRMNode(3, 5, 6, False, False)
node4 = OSRMNode(4, 7, 8, False, False)
session.add(node1)
session.add(node2)
session.add(node3)
session.add(node4)
# The source-sink nodes aren't necessarily
# in "forward" order.
edge_12 = OSRMEdge(2, 1, 5, 5, False)
edge_23 = OSRMEdge(2, 3, 5, 5, False)
edge_24 = OSRMEdge(4, 2, 5, 5, False)
session.add_all([edge_12, edge_23, edge_24])
session.commit()
# The combination of the hash + forward
# nature should define the edge frequency.
freq_12 = OSRMEdgeFrequencies(
edge=OSRMEdge.hash_edge(1, 2),
# 1->2
forward=True,
freq=12,
geom=None)
freq_21 = OSRMEdgeFrequencies(
edge=OSRMEdge.hash_edge(1, 2),
# 2->1
forward=False,
freq=21,
geom=None)
freq_34 = OSRMEdgeFrequencies(
edge=OSRMEdge.hash_edge(2, 3),
# 2-> 3
forward=True,
freq=32,
geom=None)
freq_24 = OSRMEdgeFrequencies(
edge=OSRMEdge.hash_edge(2, 4),
# This means 2->4
forward=True,
freq=24,
geom=None)
session.add_all([freq_12, freq_21, freq_34, freq_24])
session.commit()
results = session.query(OSRMEdgeFrequencies).all()
eq_(len(results), 4)
eq_(results[0].edgeobj.source, 2)
eq_(results[0].edgeobj.sink, 1)
graph = build_graph(session)
eq_(len(graph.vs), 4)
eq_(len(graph.es), 4)
assert (graph.is_directed)
eq_(set(graph.vs["osm_id"]), set([1, 2, 3, 4]))
eq_(len(graph.vs.select(osm_id=1)), 1)
vtx_1 = graph.vs.select(osm_id=1)[0]
eq_(vtx_1.indegree(), 1)
eq_(vtx_1.outdegree(), 1)
eq_(vtx_1.degree(), 2)
eq_(vtx_1.successors()[0]["osm_id"], 2)
eq_(vtx_1.predecessors()[0]["osm_id"], 2)
vtx_2 = graph.vs.select(osm_id=2)[0]
eq_(vtx_2.indegree(), 1)
eq_(vtx_2.outdegree(), 3)
eq_(vtx_2.degree(), 4)
# Check edge weights are correct
# vtx 0 is node 1
eq_(gt.output_weights(graph, 0), [12])
eq_(gt.output_weights(graph, 1), [21, 32, 24])
def test_build_graph():
with test_db_session() as session:
# Make a y-shaped graph
node1 = OSRMNode(1, 1, 2, True, False)
node2 = OSRMNode(2, 3, 4, False, False)
node3 = OSRMNode(3, 5, 6, False, False)
node4 = OSRMNode(4, 7, 8, False, False)
session.add(node1)
session.add(node2)
session.add(node3)
session.add(node4)
# The source-sink nodes aren't necessarily
# in "forward" order.
edge_12 = OSRMEdge(2, 1, 5, 5, False)
edge_23 = OSRMEdge(2, 3, 5, 5, False)
edge_24 = OSRMEdge(4, 2, 5, 5, False)
session.add_all([edge_12, edge_23, edge_24])
session.commit()
# The combination of the hash + forward
# nature should define the edge frequency.
freq_12 = OSRMEdgeFrequencies(
edge=OSRMEdge.hash_edge(1, 2),
# 1->2
forward=True,
freq=12,
geom=None
)
freq_21 = OSRMEdgeFrequencies(
edge=OSRMEdge.hash_edge(1, 2),
# 2->1
forward=False,
freq=21,
geom=None
)
freq_34 = OSRMEdgeFrequencies(
edge=OSRMEdge.hash_edge(2, 3),
# 2-> 3
forward=True,
freq=32,
geom=None
)
freq_24 = OSRMEdgeFrequencies(
edge=OSRMEdge.hash_edge(2, 4),
# This means 2->4
forward=True,
freq=24,
geom=None
)
session.add_all(
[freq_12, freq_21, freq_34, freq_24])
session.commit()
results = session.query(OSRMEdgeFrequencies).all()
eq_(len(results), 4)
eq_(results[0].edgeobj.source, 2)
eq_(results[0].edgeobj.sink, 1)
graph = build_graph(session)
eq_(len(graph.vs), 4)
eq_(len(graph.es), 4)
assert(graph.is_directed)
eq_(set(graph.vs["osm_id"]), set([1, 2, 3, 4]))
eq_(len(graph.vs.select(osm_id=1)), 1)
vtx_1 = graph.vs.select(osm_id=1)[0]
eq_(vtx_1.indegree(), 1)
eq_(vtx_1.outdegree(), 1)
eq_(vtx_1.degree(), 2)
eq_(vtx_1.successors()[0]["osm_id"], 2)
eq_(vtx_1.predecessors()[0]["osm_id"], 2)
vtx_2 = graph.vs.select(osm_id=2)[0]
eq_(vtx_2.indegree(), 1)
eq_(vtx_2.outdegree(), 3)
eq_(vtx_2.degree(), 4)
# Check edge weights are correct
# vtx 0 is node 1
eq_(gt.output_weights(graph, 0), [12])
eq_(gt.output_weights(graph, 1), [21, 32, 24])
