yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-10-31 17:59:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Malte Pietsch 2020-01-23 09:18:15 +01:00
parent d564026a57
commit 8a48cd7dd6
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_db.py
View File

@ -7,4 +7,5 @@ def test_db_write_read():
write_documents_to_db(datastore=sql_datastore, document_dir="samples/docs") write_documents_to_db(datastore=sql_datastore, document_dir="samples/docs")
documents = sql_datastore.get_all_documents() documents = sql_datastore.get_all_documents()
assert len(documents) == 2 assert len(documents) == 2
assert documents[0]["text"] == 'A Doc specifically talking about haystack.\nHaystack can be used to scale QA models to large document collections.' doc = sql_datastore.get_document_by_id("1")
assert doc["text"] == 'A Doc specifically talking about haystack.\nHaystack can be used to scale QA models to large document collections.'