mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-26 18:30:40 +00:00
20 lines
679 B
Python
20 lines
679 B
Python
import pytest
|
|
|
|
from haystack.schema import Answer
|
|
from haystack.nodes import JoinAnswers
|
|
|
|
|
|
@pytest.mark.unit
|
|
@pytest.mark.parametrize("join_mode", ["concatenate", "merge"])
|
|
def test_joinanswers(join_mode):
|
|
inputs = [{"answers": [Answer(answer="answer 1", score=0.7)]}, {"answers": [Answer(answer="answer 2", score=0.8)]}]
|
|
|
|
join_answers = JoinAnswers(join_mode=join_mode)
|
|
result, _ = join_answers.run(inputs)
|
|
assert len(result["answers"]) == 2
|
|
assert result["answers"] == sorted(result["answers"], reverse=True)
|
|
|
|
result, _ = join_answers.run(inputs, top_k_join=1)
|
|
assert len(result["answers"]) == 1
|
|
assert result["answers"][0].answer == "answer 2"
|