From cc5e68472b39333fb3e0033d30367171f2562170 Mon Sep 17 00:00:00 2001 From: vasilije Date: Wed, 27 Aug 2025 16:10:04 +0200 Subject: [PATCH] added fix to tests --- cognee/tests/unit/cli/test_cli_commands.py | 2 +- cognee/tests/unit/cli/test_cli_edge_cases.py | 18 +++++++++++++----- 2 files changed, 14 insertions(+), 6 deletions(-) diff --git a/cognee/tests/unit/cli/test_cli_commands.py b/cognee/tests/unit/cli/test_cli_commands.py index ed0ca14d1..e5d6a0c64 100644 --- a/cognee/tests/unit/cli/test_cli_commands.py +++ b/cognee/tests/unit/cli/test_cli_commands.py @@ -229,7 +229,7 @@ class TestCognifyCommand: mock_asyncio_run.assert_called_once() 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( datasets=None, diff --git a/cognee/tests/unit/cli/test_cli_edge_cases.py b/cognee/tests/unit/cli/test_cli_edge_cases.py index 0c59accda..d4e4b0ee5 100644 --- a/cognee/tests/unit/cli/test_cli_edge_cases.py +++ b/cognee/tests/unit/cli/test_cli_edge_cases.py @@ -253,7 +253,7 @@ class TestCognifyCommandEdgeCases: mock_asyncio_run.assert_called_once() 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( datasets=None, @@ -286,7 +286,7 @@ class TestCognifyCommandEdgeCases: mock_asyncio_run.assert_called_once() 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( datasets=None, @@ -306,9 +306,17 @@ class TestCognifyCommandEdgeCases: def mock_import_func(name, fromlist=None, *args, **kwargs): if name == "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") - elif name == "cognee.modules.chunking": + elif ( + name == "cognee.modules.chunking.TextChunker" + and fromlist + and "TextChunker" in fromlist + ): module = MagicMock() module.TextChunker = MagicMock() return module @@ -356,7 +364,7 @@ class TestCognifyCommandEdgeCases: mock_asyncio_run.assert_called_once() 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( datasets=None,