mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-12-30 08:37:20 +00:00
25 lines
573 B
Python
25 lines
573 B
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import pytest
|
|
|
|
from haystack.testing.sample_components import Remainder
|
|
|
|
|
|
def test_remainder_default():
|
|
component = Remainder()
|
|
results = component.run(value=4)
|
|
assert results == {"remainder_is_1": 4}
|
|
|
|
|
|
def test_remainder_with_divisor():
|
|
component = Remainder(divisor=4)
|
|
results = component.run(value=4)
|
|
assert results == {"remainder_is_0": 4}
|
|
|
|
|
|
def test_remainder_zero():
|
|
with pytest.raises(ValueError):
|
|
Remainder(divisor=0)
|