Phase 11

tests/test_face_geometry.py

@@ -0,0 +1,36 @@
+from src.config_loader import FaceSettings
+from src.face_geometry import create_face_roi
+from src.models import BoundingBox, TrackedPerson
+
+
+def settings(**overrides: object) -> FaceSettings:
+    values = dict(enabled=True, backend="mediapipe", process_every_n_frames=1, max_persons_per_frame=3,
+                  require_confirmed_track=True, process_only_inside_region=False, roi_height_ratio=.5,
+                  roi_expand_ratio=.1, min_roi_width=10, min_roi_height=10, static_image_mode=False,
+                  max_num_faces=1, refine_landmarks=True, min_detection_confidence=.5,
+                  min_tracking_confidence=.5)
+    values.update(overrides)
+    return FaceSettings(**values)  # type: ignore[arg-type]
+
+
+def person(bbox: BoundingBox, confirmed: bool = True, inside: bool = True) -> TrackedPerson:
+    return TrackedPerson(1, bbox, .9, 0, "person", bbox.center_x, bbox.center_y, bbox.foot_x, bbox.foot_y,
+                         inside, confirmed, 3, 0)
+
+
+def test_roi_ratio_expansion_and_clipping() -> None:
+    roi = create_face_roi(person(BoundingBox(10, 10, 110, 210)), 200, 300, settings())
+    assert roi is not None
+    assert (roi.x1, roi.x2) == (0, 120)
+    assert roi.y1 == 0
+    assert roi.y2 == 120
+    clipped = create_face_roi(person(BoundingBox(0, 0, 50, 100)), 60, 80, settings(roi_height_ratio=1.0))
+    assert clipped and clipped.x1 == 0 and clipped.y1 == 0 and clipped.y2 == 80
+
+
+def test_small_unconfirmed_and_outside_are_excluded() -> None:
+    assert create_face_roi(person(BoundingBox(0, 0, 20, 20)), 100, 100,
+                           settings(min_roi_width=40, min_roi_height=40)) is None
+    assert create_face_roi(person(BoundingBox(0, 0, 80, 100), confirmed=False), 100, 100, settings()) is None
+    assert create_face_roi(person(BoundingBox(0, 0, 80, 100), inside=False), 100, 100,
+                           settings(process_only_inside_region=True)) is None