18 lines
442 B
Python
18 lines
442 B
Python
import pytest
|
|
from src.nodes.rewrite import RewriteNode
|
|
|
|
|
|
def test_rewrite_node():
|
|
state = {
|
|
"reflection": (
|
|
"I see that you said: 'Hello world'. "
|
|
"Let's reflect on that."
|
|
)
|
|
}
|
|
result = RewriteNode.run(state)
|
|
assert "rewritten" in result
|
|
expected = (
|
|
"I notice that you said: 'Hello world'. "
|
|
"Let's reflect on that."
|
|
)
|
|
assert result["rewritten"] == expected |