mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-09-01 04:13:26 +00:00
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
import pytest
|
|
|
|
from haystack.preview.dataclasses import Document
|
|
from haystack.preview.testing.factory import store_class
|
|
from haystack.preview.document_stores.decorator import store
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_store_class_default():
|
|
MyStore = store_class("MyStore")
|
|
store = MyStore()
|
|
assert store.count_documents() == 0
|
|
assert store.filter_documents() == []
|
|
assert store.write_documents([]) is None
|
|
assert store.delete_documents([]) is None
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_store_class_is_registered():
|
|
MyStore = store_class("MyStore")
|
|
assert store.registry["MyStore"] == MyStore
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_store_class_with_documents():
|
|
doc = Document(id="fake_id", content="This is a document")
|
|
MyStore = store_class("MyStore", documents=[doc])
|
|
store = MyStore()
|
|
assert store.count_documents() == 1
|
|
assert store.filter_documents() == [doc]
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_store_class_with_documents_count():
|
|
MyStore = store_class("MyStore", documents_count=100)
|
|
store = MyStore()
|
|
assert store.count_documents() == 100
|
|
assert store.filter_documents() == []
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_store_class_with_documents_and_documents_count():
|
|
doc = Document(id="fake_id", content="This is a document")
|
|
MyStore = store_class("MyStore", documents=[doc], documents_count=100)
|
|
store = MyStore()
|
|
assert store.count_documents() == 100
|
|
assert store.filter_documents() == [doc]
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_store_class_with_bases():
|
|
MyStore = store_class("MyStore", bases=(Exception,))
|
|
store = MyStore()
|
|
assert isinstance(store, Exception)
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_store_class_with_extra_fields():
|
|
MyStore = store_class("MyStore", extra_fields={"my_field": 10})
|
|
store = MyStore()
|
|
assert store.my_field == 10
|