mirror of
https://github.com/enzet/map-machine.git
synced 2025-04-29 02:08:03 +02:00
c0879bff36
Element drawing module was using flinger for Mercator projection, which is not exactly what is wanted.
82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
"""
|
|
Test map generation for ways.
|
|
|
|
Tests check that for the given ways described by tags, Map Machine generates
|
|
expected figures in the expected order.
|
|
"""
|
|
import numpy as np
|
|
|
|
from map_machine.constructor import Constructor
|
|
from map_machine.figure import Figure
|
|
from map_machine.geometry.boundary_box import BoundaryBox
|
|
from map_machine.geometry.flinger import MercatorFlinger
|
|
from map_machine.map_configuration import MapConfiguration
|
|
from map_machine.osm.osm_reader import OSMData, OSMWay, OSMNode, Tags
|
|
from tests import SCHEME, SHAPE_EXTRACTOR
|
|
|
|
CONFIGURATION: MapConfiguration = MapConfiguration(SCHEME)
|
|
|
|
|
|
def get_constructor(osm_data: OSMData) -> Constructor:
|
|
"""
|
|
Get custom constructor for bounds (-0.01, -0.01, 0.01, 0.01) and zoom level
|
|
18.
|
|
"""
|
|
flinger: MercatorFlinger = MercatorFlinger(
|
|
BoundaryBox(-0.01, -0.01, 0.01, 0.01), 18, osm_data.equator_length
|
|
)
|
|
constructor: Constructor = Constructor(
|
|
osm_data, flinger, SHAPE_EXTRACTOR, CONFIGURATION
|
|
)
|
|
constructor.construct_ways()
|
|
return constructor
|
|
|
|
|
|
def create_way(osm_data: OSMData, tags: Tags, index: int) -> None:
|
|
"""Create simple OSM way with two arbitrary nodes."""
|
|
nodes: list[OSMNode] = [
|
|
OSMNode({}, 1, np.array((-0.01, -0.01))),
|
|
OSMNode({}, 2, np.array((0.01, 0.01))),
|
|
]
|
|
for node in nodes:
|
|
osm_data.add_node(node)
|
|
osm_data.add_way(OSMWay(tags, index, nodes))
|
|
|
|
|
|
def test_river_and_wood() -> None:
|
|
"""
|
|
Check that river is above the wood.
|
|
|
|
See https://github.com/enzet/map-machine/issues/126
|
|
"""
|
|
osm_data: OSMData = OSMData()
|
|
create_way(osm_data, {"natural": "wood"}, 1)
|
|
create_way(osm_data, {"waterway": "river"}, 2)
|
|
|
|
figures: list[Figure] = get_constructor(osm_data).get_sorted_figures()
|
|
|
|
assert len(figures) == 2
|
|
assert figures[0].tags["natural"] == "wood"
|
|
assert figures[1].tags["waterway"] == "river"
|
|
|
|
|
|
def test_placement_and_lanes() -> None:
|
|
"""
|
|
Check that `placement` tag is processed correctly when `lanes` tag is not
|
|
specified.
|
|
|
|
See https://github.com/enzet/map-machine/issues/128
|
|
"""
|
|
osm_data: OSMData = OSMData()
|
|
create_way(osm_data, {"highway": "motorway", "placement": "right_of:2"}, 1)
|
|
|
|
get_constructor(osm_data)
|
|
|
|
|
|
def test_empty_ways() -> None:
|
|
"""Ways without nodes."""
|
|
osm_data: OSMData = OSMData()
|
|
osm_data.add_way(OSMWay({"natural": "wood"}, 1))
|
|
osm_data.add_way(OSMWay({"waterway": "river"}, 2))
|
|
|
|
assert not get_constructor(osm_data).get_sorted_figures()
|