tests/test_state.py

from agentic_multiwriter.state import AgentState, ResearchSnippet


def test_agent_state_basic():
    state: AgentState = {
        "topic": "Test",
        "mode": "blog",
        "research_snippets": [],
        "outline": [],
        "draft": "",
        "revised_draft": "",
        "final_output": "",
        "meta": {},
    }
    assert state["topic"] == "Test"
    assert state["mode"] == "blog"