mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-29 20:00:46 +00:00
17 lines
782 B
Python
17 lines
782 B
Python
|
from haystack.components.retrievers.sentence_window_retrieval import SentenceWindowRetrieval
|
||
|
|
from haystack.components.retrievers.sentence_window_retriever import SentenceWindowRetriever
|
||
|
|
from haystack.document_stores.in_memory import InMemoryDocumentStore
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
|
||
|
|
class TestSentenceWindowRetrieval:
|
||
|
|
def test_init_default(self):
|
||
|
|
retriever = SentenceWindowRetrieval(InMemoryDocumentStore())
|
||
|
|
assert retriever.window_size == 3
|
||
|
|
|
||
|
|
def test_init_calls_parent(self):
|
||
|
|
with patch.object(SentenceWindowRetriever, "__init__", return_value=None) as mock_init:
|
||
|
|
document_store = InMemoryDocumentStore()
|
||
|
|
retriever = SentenceWindowRetrieval(document_store)
|
||
|
|
mock_init.assert_called_once_with(document_store)
|