diff --git a/cognee/tests/integration/cli/test_cli_integration.py b/cognee/tests/integration/cli/test_cli_integration.py index af7f7983a..f2e962fc8 100644 --- a/cognee/tests/integration/cli/test_cli_integration.py +++ b/cognee/tests/integration/cli/test_cli_integration.py @@ -78,7 +78,7 @@ class TestCliIntegration: [sys.executable, "-m", "cognee.cli._cognee", "add", temp_file], capture_output=True, text=True, - cwd=Path(__file__).parent.parent.parent, # Go to project root + cwd=Path(__file__).parent.parent.parent.parent, # Go to project root ) # Note: This might fail due to dependencies, but we're testing the CLI structure diff --git a/cognee/tests/unit/cli/test_cli_edge_cases.py b/cognee/tests/unit/cli/test_cli_edge_cases.py index 81d30d65f..a35afa81f 100644 --- a/cognee/tests/unit/cli/test_cli_edge_cases.py +++ b/cognee/tests/unit/cli/test_cli_edge_cases.py @@ -354,6 +354,7 @@ class TestCognifyCommandEdgeCases: command.execute(args) mock_asyncio_run.assert_called_once() + class TestDeleteCommandEdgeCases: """Test edge cases for DeleteCommand""" @@ -461,7 +462,9 @@ class TestConfigCommandEdgeCases: args = argparse.Namespace(config_action="set", key="complex_config", value=complex_json) command.execute(args) - mock_cognee.config.set.assert_called_once_with("complex_config", complex_json_expected_value) + mock_cognee.config.set.assert_called_once_with( + "complex_config", complex_json_expected_value + ) @patch("builtins.__import__") def test_config_set_invalid_json_value(self, mock_import): diff --git a/cognee/tests/unit/cli/test_cli_utils.py b/cognee/tests/unit/cli/test_cli_utils.py index b38d4adbd..00ee5f442 100644 --- a/cognee/tests/unit/cli/test_cli_utils.py +++ b/cognee/tests/unit/cli/test_cli_utils.py @@ -13,6 +13,7 @@ from cognee.cli.config import ( ) from cognee.cli._cognee import _discover_commands + class TestCliConfig: """Test CLI configuration constants""" @@ -78,6 +79,7 @@ class TestCliConfig: for expected_format in expected_formats: assert expected_format in OUTPUT_FORMAT_CHOICES + class TestCliReference: """Test CLI reference protocol"""