Final_Assignment_Template / tests /test_answer.py
Nigou Julien
Use LiteLLM for agent model calls
b22ac70
raw
history blame contribute delete
254 Bytes
from gaia_agent.answer import normalize_answer
def test_normalize_answer_strips_whitespace():
assert normalize_answer(" Paris \n") == "Paris"
def test_normalize_answer_removes_trailing_period():
assert normalize_answer("Paris.") == "Paris"