tests/test_rig.py

import numpy as np
from vid2rig.retarget.rig import load_target_rig
from vid2rig.tests.fixtures import make_mixamo_glb

def test_loads_hierarchy(tmp_path):
    p = make_mixamo_glb(tmp_path / "rig.glb")
    rig = load_target_rig(str(p))
    assert rig.bone_names == ["mixamorigHips", "mixamorigSpine", "mixamorigHead"]
    assert rig.parents == [-1, 0, 1]
    assert np.allclose(rig.bind_local_trans[1], [0, 1, 0])
    assert rig.height_m > 0