"""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()