5344cb99dd
Re-verify remaining supervision review backlog against develop HEAD; most items were already resolved by an intervening commit, only genuinely-open gaps got new fixes. Fix float32 precision loss in box_iou_batch for large coordinates (GeoTIFF-scale) by accumulating in float64. Raise ValueError instead of a strippable assert in EvaluationDataset.load_predictions for unknown image ids. Add HeatMapAnnotator.reset() to clear accumulated heat for annotator reuse. Add missing coverage: labelme export basename collisions, _greedy_match matcher, metrics.core ABC/enum contracts, metrics.utils.utils pandas guard; remove a global RNG-seed pollution site in a metrics test. Document the last two undocumented public exports (calculate_masks_centroids, is_compressed_rle) and add usage examples to 17 previously-example-less public functions/classes (NMS/NMM helpers, draw utils, PolygonZoneAnnotator, mask/polygon converters). * tests: load_predictions ValueError branch + empty-dataset coverage * fix: box_iou_batch int-dtype overflow, narrow float32 precision claim * feat: add reset() to TraceAnnotator/DetectionsSmoother, fix docstrings * docs: fix temp file leak in coco.py docstring, rename misnamed test --------- Co-authored-by: claude[bot] <209825114+claude[bot]@users.noreply.github.com> Co-authored-by: OpenAI Codex <codex@openai.com>
85 lines
3.0 KiB
Python
85 lines
3.0 KiB
Python
"""Tests for supervision.metrics.core — Metric ABC and shared enums."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from supervision.metrics.core import AveragingMethod, Metric, MetricTarget
|
|
|
|
|
|
class TestMetricAbc:
|
|
"""Verify the abstract base class cannot be instantiated directly."""
|
|
|
|
def test_cannot_instantiate_abstract_metric(self) -> None:
|
|
"""Metric is abstract and raises TypeError on direct instantiation."""
|
|
with pytest.raises(TypeError):
|
|
Metric() # type: ignore[abstract]
|
|
|
|
def test_concrete_subclass_must_implement_all_abstract_methods(self) -> None:
|
|
"""A subclass missing an abstract method still cannot be instantiated."""
|
|
|
|
class _PartialMetric(Metric):
|
|
"""Concrete class missing `reset` and `compute`."""
|
|
|
|
def update(self, *args: object, **kwargs: object) -> Metric:
|
|
"""Return self without implementing the rest of the contract."""
|
|
return self
|
|
|
|
with pytest.raises(TypeError):
|
|
_PartialMetric() # type: ignore[abstract]
|
|
|
|
def test_fully_implemented_subclass_is_instantiable(self) -> None:
|
|
"""A subclass implementing update/reset/compute can be instantiated."""
|
|
|
|
class _FullMetric(Metric):
|
|
"""Minimal concrete metric implementing the full contract."""
|
|
|
|
def update(self, *args: object, **kwargs: object) -> Metric:
|
|
"""Return self, allowing method chaining."""
|
|
return self
|
|
|
|
def reset(self) -> None:
|
|
"""No internal state to reset."""
|
|
|
|
def compute(self, *args: object, **kwargs: object) -> int:
|
|
"""Return a constant result."""
|
|
return 0
|
|
|
|
metric = _FullMetric()
|
|
assert metric.update() is metric
|
|
assert metric.compute() == 0
|
|
|
|
|
|
class TestMetricTarget:
|
|
"""Verify the MetricTarget enum members and values."""
|
|
|
|
@pytest.mark.parametrize(
|
|
("member", "value"),
|
|
[
|
|
pytest.param(MetricTarget.BOXES, "boxes", id="boxes"),
|
|
pytest.param(MetricTarget.MASKS, "masks", id="masks"),
|
|
pytest.param(
|
|
MetricTarget.ORIENTED_BOUNDING_BOXES, "obb", id="oriented-boxes"
|
|
),
|
|
],
|
|
)
|
|
def test_member_value(self, member: MetricTarget, value: str) -> None:
|
|
"""Each MetricTarget member has the documented string value."""
|
|
assert member.value == value
|
|
|
|
|
|
class TestAveragingMethod:
|
|
"""Verify the AveragingMethod enum members and values."""
|
|
|
|
@pytest.mark.parametrize(
|
|
("member", "value"),
|
|
[
|
|
pytest.param(AveragingMethod.MACRO, "macro", id="macro"),
|
|
pytest.param(AveragingMethod.MICRO, "micro", id="micro"),
|
|
pytest.param(AveragingMethod.WEIGHTED, "weighted", id="weighted"),
|
|
],
|
|
)
|
|
def test_member_value(self, member: AveragingMethod, value: str) -> None:
|
|
"""Each AveragingMethod member has the documented string value."""
|
|
assert member.value == value
|