tests/test_agents.py

from agentic_multiwriter.state import AgentState
from agentic_multiwriter.agents import aggregator_node


def test_aggregator_node_handles_empty():
    state: AgentState = {
        "topic": "Test",
        "mode": "blog",
        "research_snippets": [],
        "outline": [],
        "draft": "",
        "revised_draft": "",
        "final_output": "",
        "meta": {},
    }
    new_state = aggregator_node(state)
    assert "research_snippets" in new_state
    assert isinstance(new_state["research_snippets"], list)