This commit is contained in:
Tanay Soni 2019-11-27 19:34:10 +01:00
parent 9bf6e64529
commit d2c77f3077

View File

@ -8,7 +8,6 @@ def test_db_write_read():
db.create_all()
write_documents_to_db(document_dir="samples/docs")
documents = db.session.query(Document).all()
documents = db.session.query(Document).order_by(Document.text).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)