From 738fa9c05f864082c3cabdc7606b403dce468c3d Mon Sep 17 00:00:00 2001 From: hajdul88 <52442977+hajdul88@users.noreply.github.com> Date: Mon, 15 Dec 2025 17:41:25 +0100 Subject: [PATCH] updates ontology endpoint call in test_cognee_server_start --- cognee/tests/test_cognee_server_start.py | 4 ++-- cognee/tests/unit/api/test_ontology_endpoint.py | 4 +++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/cognee/tests/test_cognee_server_start.py b/cognee/tests/test_cognee_server_start.py index fece88240..a626088a3 100644 --- a/cognee/tests/test_cognee_server_start.py +++ b/cognee/tests/test_cognee_server_start.py @@ -148,8 +148,8 @@ class TestCogneeServerStart(unittest.TestCase): headers=headers, files=[("ontology_file", ("test.owl", ontology_content, "application/xml"))], data={ - "ontology_key": json.dumps([ontology_key]), - "description": json.dumps(["Test ontology"]), + "ontology_key": ontology_key, + "description": "Test ontology", }, ) self.assertEqual(ontology_response.status_code, 200) diff --git a/cognee/tests/unit/api/test_ontology_endpoint.py b/cognee/tests/unit/api/test_ontology_endpoint.py index 54451c86d..e072ceda8 100644 --- a/cognee/tests/unit/api/test_ontology_endpoint.py +++ b/cognee/tests/unit/api/test_ontology_endpoint.py @@ -179,7 +179,9 @@ def test_complete_multifile_workflow(client): upload_response_2 = client.post( "/api/v1/ontologies", - files=[("ontology_file", ("manufacturers.owl", io.BytesIO(file2_content), "application/xml"))], + files=[ + ("ontology_file", ("manufacturers.owl", io.BytesIO(file2_content), "application/xml")) + ], data={"ontology_key": "manufacturers", "description": "Manufacturer ontology"}, ) assert upload_response_2.status_code == 200