| """Tests for Pose2DAgent — model-dependent, skips if YOLO unavailable."""
|
| import pytest
|
| import numpy as np
|
|
|
| from formscout.types import IngestResult, Pose2DResult
|
|
|
|
|
| def _blank_ingest(n_frames=5, w=640, h=480):
|
| frames = [np.zeros((h, w, 3), dtype=np.uint8) for _ in range(n_frames)]
|
| return IngestResult(
|
| frames=frames, fps=30.0, duration=n_frames / 30.0,
|
| n_people=1, width=w, height=h,
|
| )
|
|
|
|
|
| @pytest.fixture
|
| def pose2d_agent():
|
| """Create Pose2DAgent, skip if model unavailable."""
|
| try:
|
| from formscout.agents.pose2d import Pose2DAgent
|
| agent = Pose2DAgent()
|
| return agent
|
| except Exception as e:
|
| pytest.skip(f"Pose2D model unavailable: {e}")
|
|
|
|
|
| class TestPose2DAgent:
|
| def test_returns_typed_result(self, pose2d_agent):
|
| result = pose2d_agent.run(_blank_ingest())
|
| assert isinstance(result, Pose2DResult)
|
| assert isinstance(result.keypoints, list)
|
| assert result.fps == pytest.approx(30.0)
|
|
|
| def test_keypoints_per_frame(self, pose2d_agent):
|
| ingest = _blank_ingest(n_frames=3)
|
| result = pose2d_agent.run(ingest)
|
| assert len(result.keypoints) == 3
|
| for frame_kps in result.keypoints:
|
| assert isinstance(frame_kps, dict)
|
|
|
| def test_graceful_on_empty_frames(self, pose2d_agent):
|
| empty = IngestResult(
|
| frames=[], fps=30.0, duration=0.0,
|
| n_people=0, width=640, height=480,
|
| )
|
| result = pose2d_agent.run(empty)
|
| assert result.confidence == 0.0
|
| assert "no frames" in result.notes.lower()
|
|
|
