From 8a59cadd086faf66d98ef49e39f05d7bd8b953c5 Mon Sep 17 00:00:00 2001
From: Leon Luithlen <leon@topoteretes.com>
Date: Tue, 12 Nov 2024 16:38:57 +0100
Subject: [PATCH] Remove unneeded ground truth dicts and autoformat

---
 .../tests/unit/documents/PdfDocument_test.py  |  3 +-
 .../tests/unit/interfaces/graph/conftest.py   | 36 ++-----------------
 .../graph/get_graph_from_model_test.py        |  9 ++---
 .../get_model_instance_from_graph_test.py     |  3 +-
 4 files changed, 10 insertions(+), 41 deletions(-)

diff --git a/cognee/tests/unit/documents/PdfDocument_test.py b/cognee/tests/unit/documents/PdfDocument_test.py
index 7b51e2838..86d59a561 100644
--- a/cognee/tests/unit/documents/PdfDocument_test.py
+++ b/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"},
diff --git a/cognee/tests/unit/interfaces/graph/conftest.py b/cognee/tests/unit/interfaces/graph/conftest.py
index 3f70abea2..9a784bb53 100644
--- a/cognee/tests/unit/interfaces/graph/conftest.py
+++ b/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"
diff --git a/cognee/tests/unit/interfaces/graph/get_graph_from_model_test.py b/cognee/tests/unit/interfaces/graph/get_graph_from_model_test.py
index 2311f857c..90423cc33 100644
--- a/cognee/tests/unit/interfaces/graph/get_graph_from_model_test.py
+++ b/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",
diff --git a/cognee/tests/unit/interfaces/graph/get_model_instance_from_graph_test.py b/cognee/tests/unit/interfaces/graph/get_model_instance_from_graph_test.py
index be24719d5..98ba501bd 100644
--- a/cognee/tests/unit/interfaces/graph/get_model_instance_from_graph_test.py
+++ b/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",