mirror of
https://github.com/enzet/map-machine.git
synced 2025-05-05 21:26:46 +02:00
77 lines
2.1 KiB
Python
77 lines
2.1 KiB
Python
"""
|
|
Author: Sergey Vartanov (me@enzet.ru).
|
|
"""
|
|
from os import makedirs
|
|
from typing import Dict
|
|
|
|
from roentgen.grid import draw_all_icons
|
|
from roentgen.icon import IconExtractor
|
|
from roentgen.scheme import Scheme
|
|
|
|
|
|
def test_icons() -> None:
|
|
""" Test grid drawing. """
|
|
makedirs("icon_set", exist_ok=True)
|
|
draw_all_icons("temp.svg", "icon_set")
|
|
|
|
|
|
def get_icon(tags: Dict[str, str]):
|
|
scheme = Scheme("scheme/default.yml")
|
|
icon_extractor = IconExtractor("icons/icons.svg")
|
|
icon, _ = scheme.get_icon(icon_extractor, tags)
|
|
return icon
|
|
|
|
|
|
def test_no_icons() -> None:
|
|
"""
|
|
Tags that has no description in scheme and should be visualized with default
|
|
shape.
|
|
"""
|
|
icon = get_icon({"aaa": "bbb"})
|
|
assert icon.main_icon[0].is_default()
|
|
|
|
|
|
def test_icon() -> None:
|
|
"""
|
|
Tags that should be visualized with single main icon and without extra
|
|
icons.
|
|
"""
|
|
icon = get_icon({"natural": "tree"})
|
|
assert not icon.main_icon[0].is_default()
|
|
assert not icon.extra_icons
|
|
|
|
|
|
def test_icon_1_extra() -> None:
|
|
"""
|
|
Tags that should be visualized with single main icon and single extra icon.
|
|
"""
|
|
icon = get_icon({"barrier": "gate", "access": "private"})
|
|
assert not icon.main_icon[0].is_default()
|
|
assert len(icon.extra_icons) == 1
|
|
|
|
|
|
def test_icon_2_extra() -> None:
|
|
"""
|
|
Tags that should be visualized with single main icon and two extra icons.
|
|
"""
|
|
icon = get_icon({"barrier": "gate", "access": "private", "bicycle": "yes"})
|
|
assert not icon.main_icon[0].is_default()
|
|
assert len(icon.extra_icons) == 2
|
|
|
|
|
|
def __test_no_icon_1_extra() -> None:
|
|
"""
|
|
Tags that should be visualized with default main icon and single extra icon.
|
|
"""
|
|
icon = get_icon({"access": "private"})
|
|
assert icon.main_icon[0].is_default()
|
|
assert len(icon.extra_icons) == 1
|
|
|
|
|
|
def __test_no_icon_2_extra() -> None:
|
|
"""
|
|
Tags that should be visualized with default main icon and two extra icons.
|
|
"""
|
|
icon = get_icon({"access": "private", "bicycle": "yes"})
|
|
assert icon.main_icon[0].is_default()
|
|
assert len(icon.extra_icons) == 2
|