| """Tests for the FMS session accumulator — no GPU, no model downloads."""
|
| import numpy as np
|
|
|
| from formscout.types import (
|
| IngestResult, Pose2DResult, BiomechFeatures, ScoreResult, JudgeResult,
|
| MovementResult, SessionEntry,
|
| )
|
|
|
|
|
| def test_session_entry_holds_typed_objects():
|
| movement = MovementResult(test_name="deep_squat", side="na", confidence=1.0)
|
| features = BiomechFeatures(
|
| test_name="deep_squat", view="2d", side="na",
|
| angles={"left_knee_flexion_deg": 95.0}, alignments={"knees_tracking_over_feet": True},
|
| symmetry_delta=None, timing={"deepest_frame": 2}, confidence=0.9,
|
| )
|
| rubric = ScoreResult(score=2, rationale="ok", confidence=0.8)
|
| judge = JudgeResult(score=2, rationale="ok", compensation_tags=["heels elevated"],
|
| corrective_hint="ankle mobility", confidence=0.85)
|
| entry = SessionEntry(
|
| test_name="deep_squat", side="na", score=2, needs_human=False,
|
| rationale="ok", compensation_tags=["heels elevated"], corrective_hint="ankle mobility",
|
| measurements={"left_knee_flexion_deg": 95.0}, confidence=0.85, view="2d",
|
| keyframe_path=None, movement=movement, features=features,
|
| rubric_score=rubric, judge=judge,
|
| )
|
| assert entry.score == 2
|
| assert entry.movement.test_name == "deep_squat"
|
| assert entry.rubric_score.score == 2
|
| assert entry.judge.compensation_tags == ["heels elevated"]
|
|
|
|
|
| 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 _features(test_name="deep_squat", side="na", frame_key="deepest_frame"):
|
| return BiomechFeatures(
|
| test_name=test_name, view="2d", side=side,
|
| angles={"left_knee_flexion_deg": 95.0},
|
| alignments={"knees_tracking_over_feet": False},
|
| symmetry_delta=None, timing={frame_key: 2}, confidence=0.9,
|
| )
|
|
|
|
|
| def _judge(score=2, needs_human=False):
|
| return JudgeResult(
|
| score=None if needs_human else score, rationale="r",
|
| compensation_tags=["heels elevated"], corrective_hint="ankle mobility",
|
| confidence=0.85, needs_human=needs_human,
|
| )
|
|
|
|
|
| def test_add_analysis_appends_entry_and_writes_files():
|
| import os
|
| from formscout import session as S
|
| sess = S.new_session()
|
| entry = S.add_analysis(sess, ingest=_ingest(), pose2d=_pose(),
|
| features=_features(), judge=_judge(), test_name="deep_squat", side="na")
|
| assert len(sess.entries) == 1
|
| assert entry.score == 2
|
| assert os.path.exists(os.path.join(sess.session_dir, "session.json"))
|
| assert os.path.exists(os.path.join(sess.session_dir, "analysis.md"))
|
|
|
| assert entry.keyframe_path and os.path.exists(entry.keyframe_path)
|
|
|
|
|
| def test_finish_composite_null_when_needs_human():
|
| from formscout import session as S
|
| sess = S.new_session()
|
| S.add_analysis(sess, ingest=_ingest(), pose2d=_pose(), features=_features(),
|
| judge=_judge(score=3), test_name="deep_squat", side="na")
|
| S.add_analysis(sess, ingest=_ingest(), pose2d=_pose(),
|
| features=_features("trunk_stability_pushup", frame_key="max_sag_frame"),
|
| judge=_judge(needs_human=True), test_name="trunk_stability_pushup", side="na")
|
| report, pdf_path = S.finish_session(sess)
|
| assert report is not None
|
| assert report.composite is None
|
|
|
|
|
| def test_finish_empty_session_returns_none():
|
| from formscout import session as S
|
| sess = S.new_session()
|
| report, pdf_path = S.finish_session(sess)
|
| assert report is None and pdf_path is None
|
|
|
