From 17a0e8507fa5351f83714eb7256f02ce9fc7d6ff Mon Sep 17 00:00:00 2001 From: Sergey Vartanov Date: Sat, 29 May 2021 14:27:04 +0300 Subject: [PATCH] Add reading node from XML test. --- test/test_osm_reader.py | 42 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 test/test_osm_reader.py diff --git a/test/test_osm_reader.py b/test/test_osm_reader.py new file mode 100644 index 0000000..deb78f3 --- /dev/null +++ b/test/test_osm_reader.py @@ -0,0 +1,42 @@ +""" +Test OSM XML parsing. +""" +import numpy as np +from roentgen.osm_reader import OSMNode, OSMReader, OSMRelation, OSMWay + + +def test_node() -> None: + """ + Test OSM node parsing from XML. + """ + reader = OSMReader() + map_ = reader.parse_osm_text( + """ + + +""" + ) + assert 42 in map_.node_map + node: OSMNode = map_.node_map[42] + assert node.id_ == 42 + assert np.allclose(node.coordinates, np.array([10, 5])) + + +def test_node_with_tag() -> None: + """ + Test OSM node parsing from XML. + """ + reader = OSMReader() + map_ = reader.parse_osm_text( + """ + + + + +""" + ) + assert 42 in map_.node_map + node: OSMNode = map_.node_map[42] + assert node.id_ == 42 + assert np.allclose(node.coordinates, np.array([10, 5])) + assert node.tags["key"] == "value"