from argus_mcp.models import MarketQuote
from argus_mcp.regime import build_regime_snapshot


def test_regime_prefers_risk_on_when_qqq_leads():
    snapshot = build_regime_snapshot(
        [
            MarketQuote(symbol="QQQ", source="test", change_pct=2.0),
            MarketQuote(symbol="SPY", source="test", change_pct=0.5),
            MarketQuote(symbol="BTCUSD", source="test", change_pct=1.0),
        ]
    )

    assert snapshot.regime == "risk_on"
    assert snapshot.confidence > 0


def test_regime_handles_no_data():
    snapshot = build_regime_snapshot([])

    assert snapshot.regime == "no_data"
    assert snapshot.confidence == 0.0