ci: add tests for text input/output components (#3649)

* ci: add tests for text input/output components

* ci: add tests for text input/output components

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Nicolò Boschi 2024-09-02 16:33:30 +02:00 committed by GitHub
commit 9368aef91e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 50 additions and 2 deletions

View file

@ -0,0 +1,21 @@
from langflow.schema.message import Message
from tests.integration.utils import run_single_component
from langflow.components.inputs import TextInputComponent
import pytest
@pytest.mark.asyncio
async def test_text_input():
outputs = await run_single_component(TextInputComponent, run_input="sample text", input_type="text")
print(outputs)
assert isinstance(outputs["text"], Message)
assert outputs["text"].text == "sample text"
assert outputs["text"].sender is None
assert outputs["text"].sender_name is None
outputs = await run_single_component(TextInputComponent, run_input="", input_type="text")
assert isinstance(outputs["text"], Message)
assert outputs["text"].text == ""
assert outputs["text"].sender is None
assert outputs["text"].sender_name is None

View file

@ -0,0 +1,23 @@
from langflow.components.outputs import TextOutputComponent
from langflow.schema.message import Message
from tests.integration.utils import run_single_component
import pytest
@pytest.mark.asyncio
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
@pytest.mark.asyncio
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