mirror of
https://github.com/enzet/map-machine.git
synced 2025-04-29 18:27:19 +02:00
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""Test label generation for nodes."""
|
|
from typing import Dict, List, Set
|
|
|
|
from map_machine.map_configuration import LabelMode
|
|
from map_machine.text import Label, TextConstructor
|
|
from tests import SCHEME
|
|
|
|
__author__ = "Sergey Vartanov"
|
|
__email__ = "me@enzet.ru"
|
|
|
|
|
|
def construct_labels(tags: Dict[str, str]) -> List[Label]:
|
|
"""Construct labels from OSM node tags."""
|
|
processed: Set[str] = set()
|
|
text_constructor: TextConstructor = TextConstructor(SCHEME)
|
|
return text_constructor.construct_text(tags, processed, LabelMode.ALL)
|
|
|
|
|
|
def test_1_label() -> None:
|
|
"""Test tags that should be converted into single label."""
|
|
labels = construct_labels({"name": "Name"})
|
|
assert len(labels) == 1
|
|
assert labels[0].text == "Name"
|
|
|
|
|
|
def test_1_label_unknown_tags() -> None:
|
|
"""
|
|
Test tags with some unknown tags that should be converted into single label.
|
|
"""
|
|
labels = construct_labels({"name": "Name", "aaa": "bbb"})
|
|
assert len(labels) == 1
|
|
assert labels[0].text == "Name"
|
|
|
|
|
|
def test_2_labels() -> None:
|
|
"""Test tags that should be converted into two labels."""
|
|
labels = construct_labels({"name": "Name", "ref": "5"})
|
|
assert len(labels) == 2
|
|
assert labels[0].text == "Name"
|
|
assert labels[1].text == "5"
|