Spaces:
Sleeping
Sleeping
File size: 3,804 Bytes
9f084b2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 | """Tests for geometry/normalization.py — format conversions."""
from __future__ import annotations
import pytest
from src.app.geometry import normalization
class TestXyxyToXywh:
def test_normal(self) -> None:
result = normalization.xyxy_to_xywh((10, 20, 110, 70))
assert result == (10, 20, 100, 50)
def test_swapped_coords(self) -> None:
"""Some providers may swap x1/x2 or y1/y2 — we handle it."""
result = normalization.xyxy_to_xywh((110, 70, 10, 20))
assert result == (10, 20, 100, 50)
def test_zero_size(self) -> None:
result = normalization.xyxy_to_xywh((10, 10, 10, 10))
assert result == (10, 10, 0, 0)
class TestXywhToXyxy:
def test_normal(self) -> None:
result = normalization.xywh_to_xyxy((10, 20, 100, 50))
assert result == (10, 20, 110, 70)
def test_roundtrip(self) -> None:
original = (10, 20, 110, 70)
xywh = normalization.xyxy_to_xywh(original)
back = normalization.xywh_to_xyxy(xywh)
assert back == original
class TestCxcywhToXywh:
def test_normal(self) -> None:
result = normalization.cxcywh_to_xywh((50, 50, 100, 80))
assert result == (0, 10, 100, 80)
def test_roundtrip(self) -> None:
original = (10, 20, 100, 50)
cxcywh = normalization.xywh_to_cxcywh(original)
back = normalization.cxcywh_to_xywh(cxcywh)
assert back == pytest.approx(original)
class TestFourPointToXywh:
def test_axis_aligned(self) -> None:
points = [[100, 200], [400, 200], [400, 250], [100, 250]]
result = normalization.four_point_to_xywh(points)
assert result == (100, 200, 300, 50)
def test_rotated_quad(self) -> None:
# Slightly rotated — bbox is the enclosing axis-aligned box
points = [[98, 205], [402, 195], [404, 245], [100, 255]]
result = normalization.four_point_to_xywh(points)
assert result[0] == 98 # min x
assert result[1] == 195 # min y
assert result[2] == pytest.approx(306) # max_x - min_x
assert result[3] == pytest.approx(60) # max_y - min_y
def test_tuples(self) -> None:
points = [(0, 0), (10, 0), (10, 10), (0, 10)]
result = normalization.four_point_to_xywh(points)
assert result == (0, 0, 10, 10)
def test_wrong_count(self) -> None:
with pytest.raises(ValueError, match="4 points"):
normalization.four_point_to_xywh([[0, 0], [1, 1]])
class TestFourPointToPolygon:
def test_normal(self) -> None:
points = [[100, 200], [400, 200], [400, 250], [100, 250]]
result = normalization.four_point_to_polygon(points)
assert result == [(100, 200), (400, 200), (400, 250), (100, 250)]
def test_wrong_count(self) -> None:
with pytest.raises(ValueError, match="4 points"):
normalization.four_point_to_polygon([[0, 0]])
class TestNormalizedBbox:
def test_to_pixels(self) -> None:
result = normalization.normalize_bbox_to_pixels(
(0.1, 0.2, 0.5, 0.3), 1000, 2000
)
assert result == (100, 400, 500, 600)
def test_to_normalized(self) -> None:
result = normalization.pixels_to_normalized_bbox(
(100, 400, 500, 600), 1000, 2000
)
assert result == (0.1, 0.2, 0.5, 0.3)
def test_zero_image_rejected(self) -> None:
with pytest.raises(ValueError, match="must be > 0"):
normalization.pixels_to_normalized_bbox((0, 0, 10, 10), 0, 100)
def test_roundtrip(self) -> None:
original = (50, 100, 200, 300)
norm = normalization.pixels_to_normalized_bbox(original, 1000, 2000)
back = normalization.normalize_bbox_to_pixels(norm, 1000, 2000)
assert back == pytest.approx(original)
|