langflow/src/backend/tests/integration/components/outputs/test_text_output.py
2024-11-02 13:46:07 +00:00

20 lines
774 B
Python

from langflow.components.outputs import TextOutputComponent
from langflow.schema.message import Message
from tests.integration.utils import run_single_component
async def test():
outputs = await run_single_component(TextOutputComponent, inputs={"input_value": "hello"})
assert isinstance(outputs["text"], Message)
assert outputs["text"].text == "hello"
assert outputs["text"].sender is None
assert outputs["text"].sender_name is None
async def test_message():
outputs = await run_single_component(TextOutputComponent, inputs={"input_value": Message(text="hello")})
assert isinstance(outputs["text"], Message)
assert outputs["text"].text == "hello"
assert outputs["text"].sender is None
assert outputs["text"].sender_name is None