tests/test_agent.py

from gaia_agent.agent import normalize_answer


def test_normalize_answer_removes_course_forbidden_prefix() -> None:
    assert normalize_answer("FINAL ANSWER: right") == "right"


def test_normalize_answer_unwraps_code_fence() -> None:
    assert normalize_answer("```text\na, b, c\n```") == "a, b, c"