38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib.util
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
|
|
SCRIPT = Path(__file__).parents[2] / "scripts" / "assess_belgium_building_training_iteration.py"
|
|
SPEC = importlib.util.spec_from_file_location("iteration_assessment", SCRIPT)
|
|
assert SPEC and SPEC.loader
|
|
MODULE = importlib.util.module_from_spec(SPEC)
|
|
SPEC.loader.exec_module(MODULE)
|
|
|
|
|
|
def test_calibration_selection_prefers_worst_region_then_aggregate() -> None:
|
|
report = {
|
|
"sweeps": [
|
|
{"threshold": 0.1, "pure_empty_false_positives": 0, "aggregate": {"f1": 0.8}, "regions": {"a": {"f1": 0.2}}},
|
|
{"threshold": 0.2, "pure_empty_false_positives": 0, "aggregate": {"f1": 0.6}, "regions": {"a": {"f1": 0.5}}},
|
|
]
|
|
}
|
|
assert MODULE.select_calibration_threshold(report)["threshold"] == 0.2
|
|
|
|
|
|
def test_threshold_lookup_is_exact() -> None:
|
|
report = {"sweeps": [{"threshold": 0.25, "aggregate": {}}]}
|
|
assert MODULE.find_threshold(report, 0.25)["threshold"] == 0.25
|
|
|
|
|
|
def test_release_assessment_rejects_changed_inference_configuration() -> None:
|
|
calibration = {field: None for field in MODULE.INFERENCE_CONFIG_FIELDS}
|
|
calibration.update({"model": "/models/candidate.pt", "nms_iou": 0.3, "containment_nms": 0.95})
|
|
test = dict(calibration)
|
|
test["nms_iou"] = 0.4
|
|
with pytest.raises(ValueError, match="nms_iou"):
|
|
MODULE.assert_same_inference_config(calibration, test, "test")
|