74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
from unittest.mock import Mock
|
|
from xml.etree import ElementTree as ET
|
|
|
|
import pytest
|
|
from shapely.geometry import box
|
|
from shapely.geometry.base import BaseGeometry
|
|
|
|
from flight_map.curves import interpolate
|
|
from flight_map.plot import SVG_NS, outside_index
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"points",
|
|
[
|
|
[(0, 50), (100, 50)],
|
|
[(50, 0), (50, 100)],
|
|
[(20, 30), (60, 40)],
|
|
[(0, 0), (100, 100)],
|
|
],
|
|
)
|
|
@pytest.mark.parametrize("text", ["1", "12", "123"])
|
|
def test_indices_are_outside_map_and_inside_page(
|
|
points: list[tuple[float, float]],
|
|
text: str,
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
group = ET.Element("g")
|
|
map_rectangle = box(10, 10, 90, 90)
|
|
outer_rectangle = box(4, 4, 96, 96)
|
|
occupied: list[BaseGeometry] = []
|
|
draw = Mock()
|
|
monkeypatch.setattr("flight_map.plot.label", draw)
|
|
outside_index(
|
|
group, text, interpolate(points), map_rectangle, outer_rectangle, occupied, 1.8, 0.3
|
|
)
|
|
assert len(occupied) == 1
|
|
assert occupied[0].disjoint(map_rectangle)
|
|
assert outer_rectangle.covers(occupied[0])
|
|
draw.assert_called_once()
|
|
assert group.find(f"{{{SVG_NS}}}path") is not None # Vector leader line.
|
|
|
|
|
|
def test_nearby_indices_do_not_overlap(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr("flight_map.plot.label", Mock())
|
|
occupied: list[BaseGeometry] = []
|
|
curves = interpolate([(0, 50), (100, 50)])
|
|
for index in range(1, 7):
|
|
outside_index(
|
|
ET.Element("g"),
|
|
str(index),
|
|
curves,
|
|
box(10, 10, 90, 90),
|
|
box(6, 6, 94, 94),
|
|
occupied,
|
|
1.8,
|
|
0.3,
|
|
)
|
|
for index, extent in enumerate(occupied):
|
|
assert all(extent.disjoint(other) for other in occupied[index + 1 :])
|
|
|
|
|
|
def test_too_narrow_margin_fails_instead_of_putting_index_inside() -> None:
|
|
with pytest.raises(ValueError, match="outside the map"):
|
|
outside_index(
|
|
ET.Element("g"),
|
|
"12",
|
|
interpolate([(0, 50), (100, 50)]),
|
|
box(10, 10, 90, 90),
|
|
box(9.9, 9.9, 90.1, 90.1),
|
|
[],
|
|
1.8,
|
|
0.3,
|
|
)
|