From b7d5bf5e9c11cd9822e04706699153e5a4e2f98c Mon Sep 17 00:00:00 2001 From: vasilije Date: Sun, 11 Jan 2026 16:22:06 +0100 Subject: [PATCH] test: update cognify test assertions to include chunks_per_batch parameter MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Update all cognify test assertions to expect the chunks_per_batch=None parameter that was added to the CLI command. This fixes three failing tests: - test_execute_basic_cognify - test_cognify_invalid_chunk_size - test_cognify_nonexistent_ontology_file 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Sonnet 4.5 --- cognee/tests/cli_tests/cli_unit_tests/test_cli_commands.py | 1 + cognee/tests/cli_tests/cli_unit_tests/test_cli_edge_cases.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/cognee/tests/cli_tests/cli_unit_tests/test_cli_commands.py b/cognee/tests/cli_tests/cli_unit_tests/test_cli_commands.py index 7654a781a..1301a8eaa 100644 --- a/cognee/tests/cli_tests/cli_unit_tests/test_cli_commands.py +++ b/cognee/tests/cli_tests/cli_unit_tests/test_cli_commands.py @@ -238,6 +238,7 @@ class TestCognifyCommand: ontology_file_path=None, chunker=TextChunker, run_in_background=False, + chunks_per_batch=None, ) @patch("cognee.cli.commands.cognify_command.asyncio.run") diff --git a/cognee/tests/cli_tests/cli_unit_tests/test_cli_edge_cases.py b/cognee/tests/cli_tests/cli_unit_tests/test_cli_edge_cases.py index ae9e99ac6..466a9e458 100644 --- a/cognee/tests/cli_tests/cli_unit_tests/test_cli_edge_cases.py +++ b/cognee/tests/cli_tests/cli_unit_tests/test_cli_edge_cases.py @@ -262,6 +262,7 @@ class TestCognifyCommandEdgeCases: ontology_file_path=None, chunker=TextChunker, run_in_background=False, + chunks_per_batch=None, ) @patch("cognee.cli.commands.cognify_command.asyncio.run", side_effect=_mock_run) @@ -295,6 +296,7 @@ class TestCognifyCommandEdgeCases: ontology_file_path="/nonexistent/path/ontology.owl", chunker=TextChunker, run_in_background=False, + chunks_per_batch=None, ) @patch("cognee.cli.commands.cognify_command.asyncio.run")