| from __future__ import annotations |
|
|
| import unittest |
|
|
| from src.localization_quality import ( |
| calculate_bbox_area_ratio, |
| evaluate_localization_boxes, |
| ) |
| from src.schemas.detection import BBox as LocalizationBox |
|
|
|
|
| def make_box(box_2d: list[int], label: str = "finding") -> LocalizationBox: |
| return LocalizationBox(box_2d=box_2d, label=label) |
|
|
|
|
| class TestLocalizationQualityGate(unittest.TestCase): |
| def test_a_small_focal_nodule_box_accepted(self) -> None: |
| box = make_box([388, 553, 403, 575], label="nodule") |
| result = evaluate_localization_boxes("nodule", [box]) |
|
|
| self.assertEqual(len(result.accepted_boxes), 1) |
| self.assertEqual(result.rejection_reasons, []) |
| self.assertAlmostEqual(calculate_bbox_area_ratio(box), 0.0003, places=4) |
|
|
| def test_b_broad_fracture_box_rejected(self) -> None: |
| box = make_box([100, 100, 700, 500], label="fracture") |
| result = evaluate_localization_boxes("fracture", [box]) |
|
|
| self.assertEqual(result.accepted_boxes, []) |
| self.assertTrue( |
| any("bbox_area_ratio=0.2400" in reason for reason in result.rejection_reasons) |
| ) |
|
|
| def test_c_large_diffuse_opacity_box_not_geometry_rejected(self) -> None: |
| box = make_box([380, 250, 750, 750], label="opacity") |
| result = evaluate_localization_boxes("opacity", [box]) |
|
|
| self.assertEqual(len(result.accepted_boxes), 1) |
| self.assertEqual(result.rejection_reasons, []) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
