feat: solution for 'Повторный экзамен: Граф с рефлексией и доработкой'

tests/test_graph.py

@@ -0,0 +1,14 @@
+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