tests/check_keyframes.py

"""Check keyframe counts in library moves."""
import sys
sys.path.append(".")

from reachy_mini_danceml.dataset_loader import MoveLibrary

lib = MoveLibrary()
lib.load()

# Check a few moves
sample_moves = ["a_firm_categorical_no", "a_robotic_grid_snapping", "you_look_around_use"]

for name in sample_moves:
    record = lib.get_move(name)
    if record:
        print(f"\n{name}:")
        print(f"  Keyframes: {len(record.keyframes)}")
        print(f"  Duration: {record.keyframes[-1].t:.2f}s")
        print(f"  Rate: {len(record.keyframes) / record.keyframes[-1].t:.1f} keyframes/sec")
    else:
        print(f"\n{name}: NOT FOUND")