34 lines
948 B
Python
34 lines
948 B
Python
|
"""Tests for the hermes_cli models module."""
|
||
|
|
|
||
|
|
from hermes_cli.models import OPENROUTER_MODELS, menu_labels, model_ids
|
||
|
|
|
||
|
|
|
||
|
|
class TestModelIds:
|
||
|
|
def test_returns_strings(self):
|
||
|
|
ids = model_ids()
|
||
|
|
assert isinstance(ids, list)
|
||
|
|
assert len(ids) > 0
|
||
|
|
assert all(isinstance(mid, str) for mid in ids)
|
||
|
|
|
||
|
|
def test_ids_match_models_list(self):
|
||
|
|
ids = model_ids()
|
||
|
|
expected = [mid for mid, _ in OPENROUTER_MODELS]
|
||
|
|
assert ids == expected
|
||
|
|
|
||
|
|
|
||
|
|
class TestMenuLabels:
|
||
|
|
def test_same_length_as_model_ids(self):
|
||
|
|
labels = menu_labels()
|
||
|
|
ids = model_ids()
|
||
|
|
assert len(labels) == len(ids)
|
||
|
|
|
||
|
|
def test_recommended_in_first(self):
|
||
|
|
labels = menu_labels()
|
||
|
|
assert "recommended" in labels[0].lower()
|
||
|
|
|
||
|
|
def test_labels_contain_model_ids(self):
|
||
|
|
labels = menu_labels()
|
||
|
|
ids = model_ids()
|
||
|
|
for label, mid in zip(labels, ids):
|
||
|
|
assert mid in label
|