tests/test_keyframe.py

"""Tests for PoseVisualizer.render_frame — single annotated still."""
import os
import numpy as np

from formscout.types import IngestResult, Pose2DResult


def _ingest(n=5, h=480, w=640):
    frames = [np.zeros((h, w, 3), dtype=np.uint8) for _ in range(n)]
    return IngestResult(frames=frames, fps=30.0, duration=n / 30.0, n_people=1, width=w, height=h)


def _pose(n=5):
    kps = []
    for i in range(n):
        kps.append({j: {"x": float(50 + j * 25), "y": float(80 + j * 18), "conf": 0.9}
                    for j in range(17)})
    return Pose2DResult(keypoints=kps, fps=30.0, confidence=0.9)


def test_render_frame_writes_png(tmp_path):
    from formscout.agents.visualizer import PoseVisualizer
    out = str(tmp_path / "key.png")
    path = PoseVisualizer().render_frame(_ingest(), _pose(), frame_idx=2,
                                         layers={"skeleton"}, caption="Deep Squat — heels elevated",
                                         out_png=out)
    assert path == out
    assert os.path.exists(out)
    assert os.path.getsize(out) > 0


def test_render_frame_bad_index_returns_none(tmp_path):
    from formscout.agents.visualizer import PoseVisualizer
    out = str(tmp_path / "key.png")
    path = PoseVisualizer().render_frame(_ingest(n=3), _pose(n=3), frame_idx=99,
                                         layers={"skeleton"}, caption="", out_png=out)
    assert path is None