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

fix: fixes search unit test error expectation

Browse source
This commit is contained in:
hajdul88 2025-08-13 15:00:25 +02:00
parent 885f7c3f99
commit d1bfeaa0f2
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/tests/unit/modules/search/search_methods_test.py
View file

@ -5,7 +5,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
import pytest import pytest
from pylint.checkers.utils import node_type from pylint.checkers.utils import node_type
from cognee.exceptions import InvalidValueError from cognee.modules.search.exceptions import UnsupportedSearchTypeError
from cognee.modules.search.methods.search import search, specific_search from cognee.modules.search.methods.search import search, specific_search
from cognee.modules.search.types import SearchType from cognee.modules.search.types import SearchType
from cognee.modules.users.models import User from cognee.modules.users.models import User
@ -217,7 +217,7 @@ async def test_specific_search_invalid_type(mock_user):
query_type = "INVALID_TYPE" # Not a valid SearchType query_type = "INVALID_TYPE" # Not a valid SearchType
# Execute and verify # Execute and verify
with pytest.raises(InvalidValueError) as excinfo: with pytest.raises(UnsupportedSearchTypeError) as excinfo:
await specific_search(query_type, query, mock_user) await specific_search(query_type, query, mock_user)
assert "Unsupported search type" in str(excinfo.value) assert "Unsupported search type" in str(excinfo.value)