"""Tests for vanilla LM answer quality gate."""

from __future__ import annotations

from backend.mcp.vanilla_responder import is_degraded_lm_output


def test_degraded_detects_bpe_artifacts():
    assert is_degraded_lm_output("You Ġwill Ġfind Kristian Hammond")


def test_degraded_detects_leaked_json():
    assert is_degraded_lm_output('{"action":"lookup_person","arguments":{}}')


def test_degraded_detects_special_tokens():
    assert is_degraded_lm_output("Hello <|assistant|> world")


def test_clean_answer_passes():
    assert not is_degraded_lm_output("Kristian Hammond is in Mudd Hall 3109.")


def test_empty_is_degraded():
    assert is_degraded_lm_output("")
    assert is_degraded_lm_output("   ")