gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Delete test_get_search_type_tools_integration.py

Browse source
This commit is contained in:
hajdul88 2025-12-16 15:45:32 +01:00
parent 757d5fca65
commit 48c2040f3d
1 changed files with 0 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
cognee/tests/integration/search/test_get_search_type_tools_integration.py
View file

@ -1,36 +0,0 @@
import pytest
from cognee.modules.search.types import SearchType
class _DummyCompletionContextRetriever:
def __init__(self, *args, **kwargs):
self.kwargs = kwargs
def get_completion(self, *args, **kwargs):
return None
def get_context(self, *args, **kwargs):
return None
@pytest.mark.asyncio
async def test_community_registry_is_consulted(monkeypatch):
"""
This test covers the dynamic import + lookup of community retrievers in
cognee.modules.retrieval.registered_community_retrievers.
"""
import cognee.modules.search.methods.get_search_type_tools as mod
from cognee.modules.retrieval import registered_community_retrievers as registry
monkeypatch.setattr(
registry,
"registered_community_retrievers",
{SearchType.NATURAL_LANGUAGE: _DummyCompletionContextRetriever},
)
tools = await mod.get_search_type_tools(SearchType.NATURAL_LANGUAGE, query_text="q", top_k=9)
assert len(tools) == 2
assert tools[0].__self__.kwargs["top_k"] == 9
assert tools[1].__self__.kwargs["top_k"] == 9