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_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_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_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])