roboco/tests/unit/models/test_model_map.py

"""MODEL_MAP resolves the short aliases to current, priced model ids."""

from __future__ import annotations

from roboco.billing.pricing import calculate_cost
from roboco.models.runtime import MODEL_MAP, ROLE_MODEL_MAP

_M = 1_000_000


def test_sonnet_alias_resolves_to_sonnet_5() -> None:
    assert MODEL_MAP["sonnet"] == "claude-sonnet-5"


def test_opus_alias_resolves_to_opus_4_8() -> None:
    # CEO upgrade to the newest Opus tier ("Opus 5" does not exist — the
    # Claude 5 family is Fable/Mythos/Sonnet; Opus tops out at 4.8).
    assert MODEL_MAP["opus"] == "claude-opus-4-8"


def test_sonnet_5_is_priced() -> None:
    # Guard against pointing an alias at an unpriced model (silent $0 cost-count).
    cost = calculate_cost(MODEL_MAP["sonnet"], tokens_input=_M, tokens_output=0)
    assert cost > 0.0


def test_opus_is_priced() -> None:
    # Same silent-$0 guard for the opus alias.
    cost = calculate_cost(MODEL_MAP["opus"], tokens_input=_M, tokens_output=0)
    assert cost > 0.0


def test_no_lifecycle_role_routes_below_the_structured_verb_floor() -> None:
    # Retired haiku from lifecycle roles (2026-07-24): a haiku QA/documenter
    # can't emit the structured review envelopes (criteria_verified, findings)
    # and loops without passing. No role's default may be a haiku-class tier.
    for role, model in ROLE_MODEL_MAP.items():
        assert "haiku" not in model.lower(), f"{role} routes to a below-floor {model}"


def test_main_pm_role_routes_to_sonnet() -> None:
    # Phase 2 experiment: main_pm off Opus (Sonnet 5 cache-write ~12x cheaper).
    assert ROLE_MODEL_MAP["main_pm"] == "sonnet"