mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-19 06:52:56 +00:00
5a0f0ce22f
* move functions * tests * reno * add component * reno * add tests * mypy * pylint * logger * module name
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import pytest
|
|
|
|
from haystack.components.others import Multiplexer
|
|
|
|
|
|
class TestMultiplexer:
|
|
def test_one_value(self):
|
|
multiplexer = Multiplexer(int)
|
|
output = multiplexer.run(value=[2])
|
|
assert output == {"value": 2}
|
|
|
|
def test_one_value_of_wrong_type(self):
|
|
# Multiplexer does not type check the input
|
|
multiplexer = Multiplexer(int)
|
|
output = multiplexer.run(value=["hello"])
|
|
assert output == {"value": "hello"}
|
|
|
|
def test_one_value_of_none_type(self):
|
|
# Multiplexer does not type check the input
|
|
multiplexer = Multiplexer(int)
|
|
output = multiplexer.run(value=[None])
|
|
assert output == {"value": None}
|
|
|
|
def test_more_values_of_expected_type(self):
|
|
multiplexer = Multiplexer(int)
|
|
with pytest.raises(ValueError, match="Multiplexer expects only one input, but 3 were received."):
|
|
multiplexer.run(value=[2, 3, 4])
|
|
|
|
def test_no_values(self):
|
|
multiplexer = Multiplexer(int)
|
|
with pytest.raises(ValueError, match="Multiplexer expects only one input, but 0 were received."):
|
|
multiplexer.run(value=[])
|