This commit is contained in:
Malte Pietsch 2019-11-27 17:53:42 +01:00
parent 84ce175afe
commit 7400abe327
3 changed files with 18 additions and 0 deletions

View File

@ -0,0 +1,2 @@
Some text for testing.
Two lines in here.

View File

@ -0,0 +1,2 @@
A Doc specifically talking about haystack.
Haystack can be used to scale QA models to large document collections.

14
test/test_db.py Normal file
View File

@ -0,0 +1,14 @@
from haystack.indexing.io import write_documents_to_db, fetch_archive_from_http
from haystack.database.orm import Document
def test_db_write_read():
from haystack.database import db
db.drop_all()
db.create_all()
write_documents_to_db(document_dir="samples/docs")
documents = db.session.query(Document).all()
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.'
print(documents)