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

added fix to tests

Browse source
This commit is contained in:
vasilije 2025-08-27 16:10:04 +02:00
parent 235015bf48
commit cc5e68472b
2 changed files with 14 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/tests/unit/cli/test_cli_commands.py
View file

@ -229,7 +229,7 @@ class TestCognifyCommand:
mock_asyncio_run.assert_called_once() mock_asyncio_run.assert_called_once()
assert asyncio.iscoroutine(mock_asyncio_run.call_args[0][0]) assert asyncio.iscoroutine(mock_asyncio_run.call_args[0][0])
from cognee.modules.chunking import TextChunker from cognee.modules.chunking.TextChunker import TextChunker
mock_cognee.cognify.assert_awaited_once_with( mock_cognee.cognify.assert_awaited_once_with(
datasets=None, datasets=None,

18
cognee/tests/unit/cli/test_cli_edge_cases.py
View file

@ -253,7 +253,7 @@ class TestCognifyCommandEdgeCases:
mock_asyncio_run.assert_called_once() mock_asyncio_run.assert_called_once()
assert asyncio.iscoroutine(mock_asyncio_run.call_args[0][0]) assert asyncio.iscoroutine(mock_asyncio_run.call_args[0][0])
from cognee.modules.chunking import TextChunker from cognee.modules.chunking.TextChunker import TextChunker
mock_cognee.cognify.assert_awaited_once_with( mock_cognee.cognify.assert_awaited_once_with(
datasets=None, datasets=None,
@ -286,7 +286,7 @@ class TestCognifyCommandEdgeCases:
mock_asyncio_run.assert_called_once() mock_asyncio_run.assert_called_once()
assert asyncio.iscoroutine(mock_asyncio_run.call_args[0][0]) assert asyncio.iscoroutine(mock_asyncio_run.call_args[0][0])
from cognee.modules.chunking import TextChunker from cognee.modules.chunking.TextChunker import TextChunker
mock_cognee.cognify.assert_awaited_once_with( mock_cognee.cognify.assert_awaited_once_with(
datasets=None, datasets=None,
@ -306,9 +306,17 @@ class TestCognifyCommandEdgeCases:
def mock_import_func(name, fromlist=None, *args, **kwargs): def mock_import_func(name, fromlist=None, *args, **kwargs):
if name == "cognee": if name == "cognee":
return mock_cognee return mock_cognee
elif name == "cognee.modules.chunking" and fromlist and "LangchainChunker" in fromlist: elif (
name == "cognee.modules.chunking.LangchainChunker"
and fromlist
and "LangchainChunker" in fromlist
):
raise ImportError("LangchainChunker not available") raise ImportError("LangchainChunker not available")
elif name == "cognee.modules.chunking": elif (
name == "cognee.modules.chunking.TextChunker"
and fromlist
and "TextChunker" in fromlist
):
module = MagicMock() module = MagicMock()
module.TextChunker = MagicMock() module.TextChunker = MagicMock()
return module return module
@ -356,7 +364,7 @@ class TestCognifyCommandEdgeCases:
mock_asyncio_run.assert_called_once() mock_asyncio_run.assert_called_once()
assert asyncio.iscoroutine(mock_asyncio_run.call_args[0][0]) assert asyncio.iscoroutine(mock_asyncio_run.call_args[0][0])
from cognee.modules.chunking import TextChunker from cognee.modules.chunking.TextChunker import TextChunker
mock_cognee.cognify.assert_awaited_once_with( mock_cognee.cognify.assert_awaited_once_with(
datasets=None, datasets=None,