yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-08-29 19:06:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: Check only top answer in extractive QA e2e test (#6614)

Browse Source
This commit is contained in:
Julian Risch 2023-12-22 11:11:24 +01:00 committed by GitHub
parent 8469c7f702
commit d90f95be2e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
e2e/pipelines/test_extractive_qa_pipeline.py
View File

@ -61,6 +61,10 @@ def test_extractive_qa_pipeline(tmp_path):
assert hasattr(answer, "document_offset")
assert hasattr(answer, "document")
# the answer is extracted from the correct document
if answer.document is not None:
assert answer.document.id == doc.id
# the top answer is extracted from the correct document
top_answer = extracted_answers[0]
if top_answer.document is not None:
if top_answer.document.id != doc.id:
print(top_answer.document.id, doc.id)
assert top_answer.document.id == doc.id