9194ef5abd
Docs/Test Workflow / Test docs build (push) Failing after 0s
Check links & references / links-check (push) Failing after 1s
Pytest/Test Workflow / Import Test and Pytest Run (ubuntu-latest, 3.10) (push) Failing after 0s
Pytest/Test Workflow / Import Test and Pytest Run (ubuntu-latest, 3.11) (push) Failing after 0s
PR Conflict Labeler / main (push) Failing after 2s
Pytest/Test Workflow / Import Test and Pytest Run (ubuntu-latest, 3.12) (push) Failing after 2s
Pytest/Test Workflow / Import Test and Pytest Run (ubuntu-latest, 3.13) (push) Failing after 0s
Pytest/Test Workflow / Build this Package (push) Failing after 5s
Pytest/Test Workflow / Import Test and Pytest Run (macos-latest, 3.10) (push) Has been cancelled
Pytest/Test Workflow / Import Test and Pytest Run (macos-latest, 3.11) (push) Has been cancelled
Pytest/Test Workflow / Import Test and Pytest Run (macos-latest, 3.12) (push) Has been cancelled
Pytest/Test Workflow / Import Test and Pytest Run (macos-latest, 3.13) (push) Has been cancelled
Pytest/Test Workflow / Import Test and Pytest Run (windows-latest, 3.10) (push) Has been cancelled
Pytest/Test Workflow / Import Test and Pytest Run (windows-latest, 3.11) (push) Has been cancelled
Pytest/Test Workflow / Import Test and Pytest Run (windows-latest, 3.12) (push) Has been cancelled
Pytest/Test Workflow / Import Test and Pytest Run (windows-latest, 3.13) (push) Has been cancelled
Pytest/Test Workflow / testing-guardian (push) Has been cancelled
80 lines
1.7 KiB
Python
80 lines
1.7 KiB
Python
import sys
|
|
import warnings
|
|
from pathlib import Path
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("ignore")
|
|
import matplotlib
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parent.parent
|
|
sys.path.insert(0, str(REPO_ROOT / "src"))
|
|
|
|
matplotlib.use("Agg")
|
|
|
|
import supervision as sv # noqa: E402
|
|
from tests.helpers import _create_key_points # noqa: E402
|
|
|
|
|
|
@pytest.fixture
|
|
def scene() -> np.ndarray:
|
|
return np.zeros((100, 100, 3), dtype=np.uint8)
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_key_points() -> sv.KeyPoints:
|
|
return _create_key_points(
|
|
xy=[
|
|
[
|
|
[10, 10],
|
|
[20, 20],
|
|
[30, 30],
|
|
[40, 40],
|
|
[50, 50],
|
|
[60, 60],
|
|
[70, 70],
|
|
[80, 80],
|
|
[90, 90],
|
|
[10, 20],
|
|
[20, 30],
|
|
[30, 40],
|
|
[40, 50],
|
|
[50, 60],
|
|
[60, 70],
|
|
[70, 80],
|
|
[80, 90],
|
|
],
|
|
[
|
|
[10, 40],
|
|
[20, 50],
|
|
[30, 60],
|
|
[40, 70],
|
|
[50, 80],
|
|
[60, 90],
|
|
[70, 10],
|
|
[80, 20],
|
|
[90, 30],
|
|
[10, 50],
|
|
[20, 60],
|
|
[30, 70],
|
|
[40, 80],
|
|
[50, 90],
|
|
[60, 10],
|
|
[70, 20],
|
|
[80, 30],
|
|
],
|
|
],
|
|
confidence=[
|
|
[0.8] * 17,
|
|
[0.6] * 17,
|
|
],
|
|
class_id=[0, 1],
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def empty_key_points() -> sv.KeyPoints:
|
|
return sv.KeyPoints.empty()
|