mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-10-27 15:59:14 +00:00
2e9f3c1512
* 1. Prevent update_embeddings function in FAISSDocumentStore to set faiss_index as None when document store does not have any docs. 2. cleaning up tests by adding fixture for retriever. * TfidfRetriever need document store with documents during initialization as it call fit() function in constructor so fixing it by checking self.paragraphs of None * Fix naming of retriever's fixture (embedded to embedding and tfid to tfidf)
21 lines
766 B
Python
21 lines
766 B
Python
import pytest
|
|
|
|
|
|
@pytest.mark.parametrize("document_store", ["memory"], indirect=True)
|
|
@pytest.mark.parametrize("retriever", ["tfidf"], indirect=True)
|
|
def test_tfidf_retriever(document_store, retriever):
|
|
|
|
test_docs = [
|
|
{"id": "26f84672c6d7aaeb8e2cd53e9c62d62d", "name": "testing the finder 1", "text": "godzilla says hello"},
|
|
{"name": "testing the finder 2", "text": "optimus prime says bye"},
|
|
{"name": "testing the finder 3", "text": "alien says arghh"}
|
|
]
|
|
|
|
document_store.write_documents(test_docs)
|
|
|
|
retriever.fit()
|
|
doc = retriever.retrieve("godzilla", top_k=1)[0]
|
|
assert doc.id == "26f84672c6d7aaeb8e2cd53e9c62d62d"
|
|
assert doc.text == 'godzilla says hello'
|
|
assert doc.meta == {"name": "testing the finder 1"}
|