def test_read_network_builds_graph_with_correct_data_on_nodes_and_edges():
correct_nodes = {
'21667818': {'id': '21667818', 's2_id': 5221390302696205321, 'x': 528504.1342843144, 'y': 182155.7435136598,
'lon': -0.14910908709500162, 'lat': 51.52370573323939},
'25508485': {'id': '25508485', 's2_id': 5221390301001263407, 'x': 528489.467895946, 'y': 182206.20303669578,
'lon': -0.14930198709481451, 'lat': 51.524162533239284}}
correct_edges = {'25508485_21667818': {
'id': "1", 'from': "25508485", 'to': "21667818", 'length': 52.765151087870265,
's2_from': 5221390301001263407, 's2_to': 5221390302696205321,
'freespeed': 4.166666666666667, 'capacity': 600.0, 'permlanes': 1.0, 'oneway': "1",
'modes': {'subway', 'metro', 'walk', 'car'}, 'attributes': {
'osm:way:access': {'name': 'osm:way:access', 'class': 'java.lang.String', 'text': 'permissive'},
'osm:way:highway': {'name': 'osm:way:highway', 'class': 'java.lang.String', 'text': 'unclassified'},
'osm:way:id': {'name': 'osm:way:id', 'class': 'java.lang.Long', 'text': '26997928'},
'osm:way:name': {'name': 'osm:way:name', 'class': 'java.lang.String', 'text': 'Brunswick Place'}
}}}
transformer = Transformer.from_proj(Proj('epsg:27700'), Proj('epsg:4326'), always_xy=True)
g, link_id_mapping, duplicated_nodes, duplicated_link_ids = matsim_reader.read_network(pt2matsim_network_test_file,
transformer)
for u, data in g.nodes(data=True):
assert str(u) in correct_nodes
assert_semantically_equal(data, correct_nodes[str(u)])
assert_semantically_equal(duplicated_nodes, {})
for u, v, data in g.edges(data=True):
edge = '{}_{}'.format(u, v)
assert edge in correct_edges
assert_semantically_equal(data, correct_edges[edge])
assert_semantically_equal(duplicated_link_ids, {})
def read_matsim_network(path_to_network: str, epsg: str):
"""
Reads MATSim's network.xml to genet.Network object
:param path_to_network: path to MATSim's network.xml file
:param epsg: projection for the network, e.g. 'epsg:27700'
:return: genet.Network object
"""
n = core.Network(epsg=epsg)
n.graph, n.link_id_mapping, duplicated_nodes, duplicated_links = \
matsim_reader.read_network(path_to_network, n.transformer)
n.graph.graph['name'] = 'Network graph'
n.graph.graph['crs'] = n.epsg
if 'simplified' not in n.graph.graph:
n.graph.graph['simplified'] = False
for node_id, duplicated_node_attribs in duplicated_nodes.items():
for duplicated_node_attrib in duplicated_node_attribs:
n.change_log.remove(
object_type='node',
object_id=node_id,
object_attributes=duplicated_node_attrib
)
for link_id, reindexed_duplicated_links in duplicated_links.items():
for duplicated_link in reindexed_duplicated_links:
n.change_log.modify(
object_type='link',
old_id=link_id,
old_attributes=n.link(duplicated_link),
new_id=duplicated_link,
new_attributes=n.link(duplicated_link)
)
return n
def test_read_network_builds_graph_with_unique_links_given_matsim_network_with_clashing_link_ids():
correct_nodes = {
'21667818': {'id': '21667818', 's2_id': 5221390302696205321, 'x': 528504.1342843144, 'y': 182155.7435136598,
'lon': -0.14910908709500162, 'lat': 51.52370573323939},
'25508485': {'id': '25508485', 's2_id': 5221390301001263407, 'x': 528489.467895946, 'y': 182206.20303669578,
'lon': -0.14930198709481451, 'lat': 51.524162533239284}}
correct_edges = {'25508485_21667818': {
0: {
'id': "1", 'from': "25508485", 'to': "21667818", 'length': 52.765151087870265,
's2_from': 5221390301001263407, 's2_to': 5221390302696205321,
'freespeed': 4.166666666666667, 'capacity': 600.0, 'permlanes': 1.0, 'oneway': "1",
'modes': {'walk', 'car'}, 'attributes': {
'osm:way:access': {'name': 'osm:way:access', 'class': 'java.lang.String', 'text': 'permissive'},
'osm:way:highway': {'name': 'osm:way:highway', 'class': 'java.lang.String', 'text': 'unclassified'},
'osm:way:id': {'name': 'osm:way:id', 'class': 'java.lang.Long', 'text': '26997928'},
'osm:way:name': {'name': 'osm:way:name', 'class': 'java.lang.String', 'text': 'Brunswick Place'}}
},
1: {
'id': "1_1", 'from': "25508485", 'to': "21667818", 'length': 52.765151087870265,
's2_from': 5221390301001263407, 's2_to': 5221390302696205321,
'freespeed': 4.166666666666667, 'capacity': 600.0, 'permlanes': 1.0, 'oneway': "1",
'modes': {'bus'}, 'attributes': {
'osm:way:lanes': {'name': 'osm:way:lanes', 'class': 'java.lang.String', 'text': '1'},
'osm:way:highway': {'name': 'osm:way:highway', 'class': 'java.lang.String', 'text': 'unclassified'},
'osm:way:id': {'name': 'osm:way:id', 'class': 'java.lang.Long', 'text': '26997928'},
'osm:way:name': {'name': 'osm:way:name', 'class': 'java.lang.String', 'text': 'Brunswick Place'},
'osm:way:oneway': {'name': 'osm:way:oneway', 'class': 'java.lang.String', 'text': 'yes'},
'osm:relation:route': {'class': 'java.lang.String', 'name': 'osm:relation:route',
'text': {'bus', 'bicycle'}}
}
}}}
correct_link_id_map = {'1': {'from': '25508485', 'to': '21667818', 'multi_edge_idx': 0},
'1_1': {'from': '25508485', 'to': '21667818', 'multi_edge_idx': 1}}
transformer = Transformer.from_proj(Proj('epsg:27700'), Proj('epsg:4326'), always_xy=True)
g, link_id_mapping, duplicated_nodes, duplicated_link_ids = matsim_reader.read_network(
pt2matsim_network_clashing_link_ids_test_file, transformer)
assert len(g.nodes) == len(correct_nodes)
for u, data in g.nodes(data=True):
assert str(u) in correct_nodes
assert_semantically_equal(data, correct_nodes[str(u)])
assert_semantically_equal(duplicated_nodes, {})
for edge in g.edges:
e = '{}_{}'.format(edge[0], edge[1])
assert e in correct_edges
assert edge[2] in correct_edges[e]
assert_semantically_equal(g[edge[0]][edge[1]][edge[2]], correct_edges[e][edge[2]])
assert_semantically_equal(correct_link_id_map, link_id_mapping)
assert_semantically_equal(duplicated_link_ids, {'1': ['1_1']})
