mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-07 00:51:22 +00:00
85258f0654
* Fix types in test_run.py * Get test_run.py to pass fmt-check * Add test_run to mypy checks * Update test folder to pass ruff linting * Fix merge * Fix HF tests * Fix hf test * Try to fix tests * Another attempt * minor fix * fix SentenceTransformersDiversityRanker * skip integrations tests due to model unavailable on HF inference --------- Co-authored-by: anakin87 <stefanofiorucci@gmail.com>
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from haystack.components.joiners.string_joiner import StringJoiner
|
|
from haystack.core.serialization import component_from_dict, component_to_dict
|
|
|
|
|
|
class TestStringJoiner:
|
|
def test_init(self):
|
|
joiner = StringJoiner()
|
|
assert isinstance(joiner, StringJoiner)
|
|
|
|
def test_to_dict(self):
|
|
joiner = StringJoiner()
|
|
data = component_to_dict(joiner, name="string_joiner")
|
|
assert data == {"type": "haystack.components.joiners.string_joiner.StringJoiner", "init_parameters": {}}
|
|
|
|
def test_from_dict(self):
|
|
data = {"type": "haystack.components.joiners.string_joiner.StringJoiner", "init_parameters": {}}
|
|
string_joiner = component_from_dict(StringJoiner, data=data, name="string_joiner")
|
|
assert isinstance(string_joiner, StringJoiner)
|
|
|
|
def test_empty_list(self):
|
|
joiner = StringJoiner()
|
|
result = joiner.run([])
|
|
assert result == {"strings": []}
|
|
|
|
def test_single_string(self):
|
|
joiner = StringJoiner()
|
|
result = joiner.run("a")
|
|
assert result == {"strings": ["a"]}
|
|
|
|
def test_two_strings(self):
|
|
joiner = StringJoiner()
|
|
result = joiner.run(["a", "b"])
|
|
assert result == {"strings": ["a", "b"]}
|