haystack/test/core/sample_components/test_sum.py

# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
#
# SPDX-License-Identifier: Apache-2.0

from haystack.testing.sample_components import Sum


def test_sum_receives_no_values():
    component = Sum()
    results = component.run(values=[])
    assert results == {"total": 0}


def test_sum_receives_one_value():
    component = Sum()
    assert component.run(values=[10]) == {"total": 10}


def test_sum_receives_few_values():
    component = Sum()
    assert component.run(values=[10, 2]) == {"total": 12}