tests/test_graph.py

from graph.state import initial_state
from api.schemas import GaiaQuestion


def test_initial_state():
    question = GaiaQuestion(task_id="1", question="What is 2+2?")
    state = initial_state(question)
    assert state["task_id"] == "1"
    assert state["question"] == "What is 2+2?"
    assert state["step_index"] == 0
    assert state["final_answer"] == ""