14 lines
362 B
Python
14 lines
362 B
Python
import pytest
|
|
from src.graph import build_graph
|
|
|
|
|
|
def test_graph_flow():
|
|
graph = build_graph()
|
|
input_state = {"input": "Hello world"}
|
|
result = graph.invoke(input_state)
|
|
assert "rewritten" in result
|
|
expected = (
|
|
"I notice that you said: 'Hello world'. "
|
|
"Let's reflect on that."
|
|
)
|
|
assert result["rewritten"] == expected |