13 lines
336 B
Python
13 lines
336 B
Python
import pytest
|
|
from src.nodes.reflect import ReflectNode
|
|
|
|
|
|
def test_reflect_node():
|
|
state = {"input": "Hello world"}
|
|
result = ReflectNode.run(state)
|
|
assert "reflection" in result
|
|
expected = (
|
|
"I see that you said: 'Hello world'. "
|
|
"Let's reflect on that."
|
|
)
|
|
assert result["reflection"] == expected |