mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-27 10:49:52 +00:00
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from haystack.reader.base import BaseReader
|
||
|
|
from haystack.database.base import Document
|
||
|
|
|
||
|
|
|
||
|
|
def test_reader_basic(reader):
|
||
|
|
assert reader is not None
|
||
|
|
assert isinstance(reader, BaseReader)
|
||
|
|
|
||
|
|
|
||
|
|
def test_output(reader, test_docs_xs):
|
||
|
|
docs = []
|
||
|
|
for d in test_docs_xs:
|
||
|
|
doc = Document(id=d["name"], text=d["text"], meta=d["meta"])
|
||
|
|
docs.append(doc)
|
||
|
|
results = reader.predict(question="Who lives in Berlin?", documents=docs, top_k=5)
|
||
|
|
assert results is not None
|
||
|
|
assert results["question"] == "Who lives in Berlin?"
|
||
|
|
assert results["answers"][0]["answer"] == "Carla"
|
||
|
|
assert results["answers"][0]["offset_start"] == 11
|
||
|
|
#TODO enable again when FARM is upgraded incl. the new offset calc
|
||
|
|
# assert results["answers"][0]["offset_end"] == 16
|
||
|
|
assert results["answers"][0]["probability"] <= 1
|
||
|
|
assert results["answers"][0]["probability"] >= 0
|
||
|
|
assert results["answers"][0]["context"] == "My name is Carla and I live in Berlin"
|
||
|
|
assert results["answers"][0]["document_id"] == "filename1"
|
||
|
|
assert len(results["answers"]) == 5
|