Initial public release

tests/test_audit_yolo_cross_tile_repetition.py

@@ -0,0 +1,50 @@
+import pytest
+
+from scripts.audit_yolo_cross_tile_repetition import (
+    interior_global_key,
+    tile_offsets,
+)
+from scripts.audit_yolo_label_relationships import Box
+
+
+def test_tile_offsets_parse_exporter_filename() -> None:
+    assert tile_offsets("/data/geel_0005_r256_c512.png") == (256, 512)
+    with pytest.raises(ValueError, match="no row/column offsets"):
+        tile_offsets("/data/not-a-tile.png")
+
+
+def test_interior_global_key_reconstructs_same_object_across_tiles() -> None:
+    first = Box(0, 0.75, 0.5, 0.1, 0.1)
+    second = Box(0, 0.25, 0.5, 0.1, 0.1)
+
+    first_key = interior_global_key(
+        first,
+        row_offset=0,
+        column_offset=0,
+        tile_size=512,
+        edge_tolerance_pixels=0.5,
+    )
+    second_key = interior_global_key(
+        second,
+        row_offset=0,
+        column_offset=256,
+        tile_size=512,
+        edge_tolerance_pixels=0.5,
+    )
+
+    assert first_key == second_key == (358.4, 230.4, 409.6, 281.6)
+
+
+def test_interior_global_key_rejects_edge_clipped_box() -> None:
+    edge_box = Box(0, 0.05, 0.5, 0.1, 0.1)
+
+    assert (
+        interior_global_key(
+            edge_box,
+            row_offset=0,
+            column_offset=256,
+            tile_size=512,
+            edge_tolerance_pixels=0.5,
+        )
+        is None
+    )