mirror of
https://github.com/azaion/detections.git
synced 2026-04-22 08:56:32 +00:00
[AZ-139] [AZ-140] [AZ-141] [AZ-142] Implement integration tests for health, single image, tiling, and async SSE
Made-with: Cursor
This commit is contained in:
@@ -1 +1,57 @@
|
||||
"""Large-image tiling and overlap behavior for POST /detect."""
|
||||
import json
|
||||
|
||||
import pytest
|
||||
|
||||
_TILING_TIMEOUT = 120
|
||||
_GSD = {"altitude": 400, "focal_length": 24, "sensor_width": 23.5}
|
||||
_DUP_THRESHOLD = 0.01
|
||||
|
||||
|
||||
def _assert_coords_normalized(detections):
|
||||
for d in detections:
|
||||
for k in ("centerX", "centerY", "width", "height"):
|
||||
v = d[k]
|
||||
assert 0.0 <= v <= 1.0
|
||||
|
||||
|
||||
def _assert_no_same_label_near_duplicate_centers(detections):
|
||||
by_label = {}
|
||||
for d in detections:
|
||||
label = d["label"]
|
||||
cx, cy = d["centerX"], d["centerY"]
|
||||
prev = by_label.setdefault(label, [])
|
||||
for pcx, pcy in prev:
|
||||
assert not (
|
||||
abs(cx - pcx) < _DUP_THRESHOLD and abs(cy - pcy) < _DUP_THRESHOLD
|
||||
), f"near-duplicate centers for label {label!r}: ({pcx},{pcy}) vs ({cx},{cy})"
|
||||
prev.append((cx, cy))
|
||||
|
||||
|
||||
@pytest.mark.slow
|
||||
def test_ft_p_04_gsd_based_tiling_ac1(http_client, image_large, warm_engine):
|
||||
config = json.dumps(_GSD)
|
||||
r = http_client.post(
|
||||
"/detect",
|
||||
files={"file": ("img.jpg", image_large, "image/jpeg")},
|
||||
data={"config": config},
|
||||
timeout=_TILING_TIMEOUT,
|
||||
)
|
||||
assert r.status_code == 200
|
||||
body = r.json()
|
||||
assert isinstance(body, list)
|
||||
_assert_coords_normalized(body)
|
||||
|
||||
|
||||
@pytest.mark.slow
|
||||
def test_ft_p_16_tile_boundary_deduplication_ac2(http_client, image_large, warm_engine):
|
||||
config = json.dumps({**_GSD, "big_image_tile_overlap_percent": 20})
|
||||
r = http_client.post(
|
||||
"/detect",
|
||||
files={"file": ("img.jpg", image_large, "image/jpeg")},
|
||||
data={"config": config},
|
||||
timeout=_TILING_TIMEOUT,
|
||||
)
|
||||
assert r.status_code == 200
|
||||
body = r.json()
|
||||
assert isinstance(body, list)
|
||||
_assert_no_same_label_near_duplicate_centers(body)
|
||||
|
||||
Reference in New Issue
Block a user