Files
Video-Tree-TRM5/tests/unit/test_split_selection.py
T

20 lines
631 B
Python

import pytest
from app.harness.split_selection import cell_of, evolution_target_of
def test_evolution_target_mapping():
assert evolution_target_of("extraction_failure") == "tool"
assert evolution_target_of("search_failure") == "skill"
assert evolution_target_of("reasoning_failure") == "skill"
assert evolution_target_of("mixed") == "system"
def test_evolution_target_unknown_raises():
with pytest.raises(ValueError):
evolution_target_of("unknown_type")
def test_cell_is_task_type_x_error_type():
assert cell_of("Counting Problem", "search_failure") == ("Counting Problem", "search_failure")