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