Remove unneeded ground truth dicts and autoformat

cognee/tests/unit/documents/PdfDocument_test.py

@@ -1,8 +1,7 @@
 import os
 import uuid
 
-from cognee.modules.data.processing.document_types.PdfDocument import \
-    PdfDocument
+from cognee.modules.data.processing.document_types.PdfDocument import PdfDocument
 
 GROUND_TRUTH = [
     {"word_count": 879, "len_text": 5622, "cut_type": "sentence_end"},

cognee/tests/unit/interfaces/graph/conftest.py

@@ -5,41 +5,11 @@ from typing import Optional
 import pytest
 
 from cognee.infrastructure.engine import DataPoint
-from cognee.modules.graph.utils import (get_graph_from_model,
-                                        get_model_instance_from_graph)
-
-EDGE_GROUND_TRUTH = (
-    "boris",
-    "car1",
-    "owns_car",
-    {
-        "source_node_id": "boris",
-        "target_node_id": "car1",
-        "relationship_name": "owns_car",
-        "metadata": {"type": "list"},
-    },
+from cognee.modules.graph.utils import (
+    get_graph_from_model,
+    get_model_instance_from_graph,
 )
 
-CAR_GROUND_TRUTH = {
-    "id": "car1",
-    "brand": "Toyota",
-    "model": "Camry",
-    "year": 2020,
-    "color": "Blue",
-}
-
-PERSON_GROUND_TRUTH = {
-    "id": "boris",
-    "name": "Boris",
-    "age": 30,
-    "driving_license": {
-        "issued_by": "PU Vrsac",
-        "issued_on": "2025-11-06",
-        "number": "1234567890",
-        "expires_on": "2025-11-06",
-    },
-}
-
 
 class CarTypeName(Enum):
     Pickup = "Pickup"

cognee/tests/unit/interfaces/graph/get_graph_from_model_test.py

@@ -1,8 +1,9 @@
 from cognee.infrastructure.engine import DataPoint
-from cognee.modules.graph.utils import (get_graph_from_model,
-                                        get_model_instance_from_graph)
-from cognee.tests.unit.interfaces.graph.util import \
-    run_test_against_ground_truth
+from cognee.modules.graph.utils import (
+    get_graph_from_model,
+    get_model_instance_from_graph,
+)
+from cognee.tests.unit.interfaces.graph.util import run_test_against_ground_truth
 
 EDGE_GROUND_TRUTH = (
     "boris",

cognee/tests/unit/interfaces/graph/get_model_instance_from_graph_test.py

@@ -1,5 +1,4 @@
-from cognee.tests.unit.interfaces.graph.util import \
-    run_test_against_ground_truth
+from cognee.tests.unit.interfaces.graph.util import run_test_against_ground_truth
 
 PARSED_PERSON_GROUND_TRUTH = {
     "id": "boris",