mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-28 11:19:58 +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)
|