Skip to content

Commit 2c92f1d

Browse files
committed
test(unit): add test for model without vendor
1 parent eb4449d commit 2c92f1d

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

tests/unit/test_collector.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -63,17 +63,18 @@ def test_model_with_colon_in_name(self, tmp_path: Path) -> None:
6363
assert "meta-llama" in str(result)
6464
assert "llama-3:70b" in str(result)
6565

66-
def test_invalid_model_format_raises(self, tmp_path: Path) -> None:
67-
"""Invalid model format should raise AssertionError."""
66+
def test_model_without_vendor_defaults_to_other(self, tmp_path: Path) -> None:
67+
"""Models without vendor should default to 'other'."""
6868
task = Task(
6969
model="invalid_model", # Missing vendor/
7070
seed="TEST",
7171
deck="RED",
7272
stake="WHITE",
7373
strategy="default",
7474
)
75-
with pytest.raises(AssertionError, match="Invalid vendor/model"):
76-
_generate_run_dir(task, tmp_path)
75+
result = _generate_run_dir(task, tmp_path)
76+
assert "other" in str(result)
77+
assert "invalid_model" in str(result)
7778

7879

7980
# ============================================================================

0 commit comments

Comments
 (0)