test: suite pytest sintetica (GT pose note) + deps dev pytest/ruff
11 test senza dipendenza dalle immagini Test/ (non versionate): - precisione/recall su 7 pose GT (soglie 0.2-0.5 deg, 0.3-1.0 px, margine 3-4x sulle misure Fase 2) - unit: angle_list con estremi, clamp piramide, save/load roundtrip, no collisione cache scena, mask poligonale, find non addestrato Config ruff in pyproject (E702/E402 idiomi del codebase esclusi). Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,84 @@
|
||||
"""Unit test rapidi su componenti del matcher (no matching pesante)."""
|
||||
from __future__ import annotations
|
||||
|
||||
import numpy as np
|
||||
import cv2
|
||||
import pytest
|
||||
|
||||
from pm2d import LineShapeMatcher
|
||||
from tests.conftest import GT_POSES, make_scene, match_errors
|
||||
|
||||
|
||||
def test_angle_list_includes_range_end():
|
||||
# Range parziale ±15: l'estremo +15 deve essere testato (era escluso).
|
||||
m = LineShapeMatcher(angle_range_deg=(-15.0, 15.0), angle_step_deg=5.0)
|
||||
angles = m._angle_list()
|
||||
assert -15.0 in angles and 15.0 in angles
|
||||
assert len(angles) == 7
|
||||
|
||||
|
||||
def test_angle_list_full_circle_no_duplicate():
|
||||
# (0, 360): 360 coincide con 0 → escluso, niente variante duplicata.
|
||||
m = LineShapeMatcher(angle_range_deg=(0.0, 360.0), angle_step_deg=5.0)
|
||||
angles = m._angle_list()
|
||||
assert len(angles) == 72
|
||||
assert 360.0 not in angles
|
||||
|
||||
|
||||
def test_pyramid_clamp_small_template():
|
||||
# Template 40px di lato minimo: al top /4 le feature collassano →
|
||||
# i livelli vengono clampati (40/2=20 >= 12, 40/4=10 < 12 → 2 livelli).
|
||||
m = LineShapeMatcher(pyramid_levels=4, angle_range_deg=(0.0, 10.0),
|
||||
angle_step_deg=5.0)
|
||||
tpl = np.full((40, 200), 60, np.uint8)
|
||||
cv2.rectangle(tpl, (30, 8), (170, 32), 200, -1)
|
||||
m.train(tpl)
|
||||
assert m.pyramid_levels == 2
|
||||
|
||||
|
||||
def test_save_load_roundtrip(tmp_path, template, scene):
|
||||
m = LineShapeMatcher(angle_step_deg=10.0)
|
||||
m.train(template)
|
||||
path = str(tmp_path / "model.npz")
|
||||
m.save_model(path)
|
||||
m2 = LineShapeMatcher.load_model(path)
|
||||
assert len(m2.variants) == len(m.variants)
|
||||
matches = m2.find(scene, min_score=0.5, max_matches=10)
|
||||
_, _, miss = match_errors(matches, GT_POSES)
|
||||
assert miss == 0
|
||||
|
||||
|
||||
def test_scene_cache_no_collision(template):
|
||||
# Due scene IDENTICHE nella banda superiore ma diverse sotto: la cache
|
||||
# (che prima hashava solo i primi 64KB) non deve restituire i risultati
|
||||
# della scena sbagliata.
|
||||
poses_a = [GT_POSES[0], (450.0, 560.0, 33.7)]
|
||||
poses_b = [GT_POSES[0], (700.0, 560.0, 91.2)]
|
||||
scene_a = make_scene(template, poses_a)
|
||||
scene_b = make_scene(template, poses_b)
|
||||
# Stessa banda superiore (le pose extra sono in basso, y >= 430)
|
||||
assert np.array_equal(scene_a[:80], scene_b[:80])
|
||||
m = LineShapeMatcher(angle_step_deg=10.0)
|
||||
m.train(template)
|
||||
ma = m.find(scene_a, min_score=0.5, max_matches=5)
|
||||
mb = m.find(scene_b, min_score=0.5, max_matches=5)
|
||||
_, _, miss_a = match_errors(ma, poses_a)
|
||||
_, _, miss_b = match_errors(mb, poses_b)
|
||||
assert miss_a == 0 and miss_b == 0
|
||||
|
||||
|
||||
def test_train_mask_polygonal(template, scene):
|
||||
# ROI poligonale: mask che copre solo la L verticale del template.
|
||||
mask = np.zeros_like(template)
|
||||
cv2.rectangle(mask, (10, 10), (70, template.shape[0] - 10), 255, -1)
|
||||
m = LineShapeMatcher(angle_step_deg=10.0)
|
||||
n = m.train(template, mask=mask)
|
||||
assert n > 0
|
||||
matches = m.find(scene, min_score=0.5, max_matches=10)
|
||||
assert len(matches) >= 1
|
||||
|
||||
|
||||
def test_untrained_find_raises():
|
||||
m = LineShapeMatcher()
|
||||
with pytest.raises(RuntimeError):
|
||||
m.find(np.zeros((100, 100), np.uint8))
|
||||
Reference in New Issue
Block a user