09de0f4f06
* feat(ruff): add pytest linting rule * --unsafe-fixes * fix(pre_commit): 🎨 auto format pre-commit hooks --------- Co-authored-by: jiri BOROVEC <jirka@Jiris-MacBook-Pro-J4N257Y6Q2.local> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: Copilot <198982749+Copilot@users.noreply.github.com>
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
import numpy as np
|
|
import pytest
|
|
|
|
import supervision as sv
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("detections", "expected_results"),
|
|
[
|
|
(
|
|
[
|
|
sv.Detections(
|
|
xyxy=np.array([[10, 10, 20, 20], [30, 30, 40, 40]]),
|
|
class_id=np.array([1, 1]),
|
|
confidence=np.array([1, 1]),
|
|
),
|
|
sv.Detections(
|
|
xyxy=np.array([[10, 10, 20, 20], [30, 30, 40, 40]]),
|
|
class_id=np.array([1, 1]),
|
|
confidence=np.array([1, 1]),
|
|
),
|
|
],
|
|
sv.Detections(
|
|
xyxy=np.array([[10, 10, 20, 20], [30, 30, 40, 40]]),
|
|
class_id=np.array([1, 1]),
|
|
confidence=np.array([1, 1]),
|
|
tracker_id=np.array([1, 2]),
|
|
),
|
|
),
|
|
],
|
|
)
|
|
def test_byte_tracker(
|
|
detections: list[sv.Detections],
|
|
expected_results: sv.Detections,
|
|
) -> None:
|
|
byte_tracker = sv.ByteTrack()
|
|
tracked_detections = [byte_tracker.update_with_detections(d) for d in detections]
|
|
assert tracked_detections[-1] == expected_results
|