haystack/test/core/sample_components/test_concatenate.py
2025-04-29 18:09:11 +02:00

30 lines
838 B
Python

# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
#
# SPDX-License-Identifier: Apache-2.0
from haystack.testing.sample_components import Concatenate
def test_input_lists():
component = Concatenate()
res = component.run(first=["This"], second=["That"])
assert res == {"value": ["This", "That"]}
def test_input_strings():
component = Concatenate()
res = component.run(first="This", second="That")
assert res == {"value": ["This", "That"]}
def test_input_first_list_second_string():
component = Concatenate()
res = component.run(first=["This"], second="That")
assert res == {"value": ["This", "That"]}
def test_input_first_string_second_list():
component = Concatenate()
res = component.run(first="This", second=["That"])
assert res == {"value": ["This", "That"]}