mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-21 07:51:40 +00:00
12 lines
463 B
Python
12 lines
463 B
Python
from haystack.database.sql import SQLDocumentStore
|
|
from haystack.indexing.io import write_documents_to_db
|
|
|
|
|
|
def test_db_write_read():
|
|
sql_document_store = SQLDocumentStore()
|
|
write_documents_to_db(document_store=sql_document_store, document_dir="samples/docs")
|
|
documents = sql_document_store.get_all_documents()
|
|
assert len(documents) == 2
|
|
doc = sql_document_store.get_document_by_id("1")
|
|
assert doc.keys() == {"id", "name", "text", "tags"}
|