haystack/test/core/sample_components/test_threshold.py
Massimiliano Pippi 00e1dd6eb8
chore: rearrange the core package, move tests and clean up (#6427)
* rearrange code

* fix tests

* relnote

* merge test modules

* remove extra

* rearrange draw tests

* forgot

* remove unused import
2023-11-28 09:58:56 +01:00

16 lines
474 B
Python

# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
#
# SPDX-License-Identifier: Apache-2.0
from haystack.testing.sample_components import Threshold
from haystack.core.serialization import component_to_dict, component_from_dict
def test_threshold():
component = Threshold()
results = component.run(value=5, threshold=10)
assert results == {"below": 5}
results = component.run(value=15, threshold=10)
assert results == {"above": 15}