iomgaa/Video-Tree-TRM5
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(tests): update test_generator_v2 to use new generator signatures

Browse Source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
iomgaa
2026-07-14 05:55:08 -04:00
parent afa77173e3
commit b9616de21e
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/test_generator_v2.py
+14 -6
View File
@@ -165,7 +165,8 @@ class TestBuildV2Prompt:
material = _make_material()
messages, frame_paths = _build_v2_prompt(
family_spec=RETRIEVAL_FAMILY,
prompt_template=RETRIEVAL_FAMILY.prompt_template,
strategy_name=RETRIEVAL_FAMILY.name,
material=material,
task_type="Action Reasoning",
seq=1,
@@ -184,7 +185,8 @@ class TestBuildV2Prompt:
material = _make_material()
reject_msg = "The question leaked information from the answer options."
messages, _ = _build_v2_prompt(
family_spec=RETRIEVAL_FAMILY,
prompt_template=RETRIEVAL_FAMILY.prompt_template,
strategy_name=RETRIEVAL_FAMILY.name,
material=material,
task_type="Action Reasoning",
seq=2,
@@ -201,7 +203,8 @@ class TestBuildV2Prompt:
material = _make_material(with_frames=True)
_, frame_paths = _build_v2_prompt(
family_spec=VISUAL_FAMILY,
prompt_template=VISUAL_FAMILY.prompt_template,
strategy_name=VISUAL_FAMILY.name,
material=material,
task_type="Object Recognition",
seq=1,
@@ -215,7 +218,8 @@ class TestBuildV2Prompt:
material = _make_material(with_frames=False)
_, frame_paths = _build_v2_prompt(
family_spec=RETRIEVAL_FAMILY,
prompt_template=RETRIEVAL_FAMILY.prompt_template,
strategy_name=RETRIEVAL_FAMILY.name,
material=material,
task_type="Action Reasoning",
seq=1,
@@ -355,10 +359,12 @@ class TestGenerateOneV2:
vlm=mock_vlm,
tree=tree,
material=material,
family_spec=RETRIEVAL_FAMILY,
task_type="Action Reasoning",
seq=1,
video_id="v-001",
prompt_template=RETRIEVAL_FAMILY.prompt_template,
strategy_name=RETRIEVAL_FAMILY.name,
skill_target=RETRIEVAL_FAMILY.skill_target,
session_id="session-test-001",
)
@@ -387,10 +393,12 @@ class TestGenerateOneV2:
vlm=mock_vlm,
tree=tree,
material=material,
family_spec=RETRIEVAL_FAMILY,
task_type="Action Reasoning",
seq=2,
video_id="v-001",
prompt_template=RETRIEVAL_FAMILY.prompt_template,
strategy_name=RETRIEVAL_FAMILY.name,
skill_target=RETRIEVAL_FAMILY.skill_target,
reject_reason="Answer was too obvious",
session_id="session-test-002",
)