From 7a6cf5329247b083fbf62cc9e378b4ccf4a00bb3 Mon Sep 17 00:00:00 2001
From: Leon Luithlen <leon@topoteretes.com>
Date: Tue, 12 Nov 2024 14:52:24 +0100
Subject: [PATCH] Remove tests/unit/integration folder

---
 .../test_model_to_graph_to_model.py           | 171 ------------------
 1 file changed, 171 deletions(-)
 delete mode 100644 cognee/tests/unit/integration/test_model_to_graph_to_model.py

diff --git a/cognee/tests/unit/integration/test_model_to_graph_to_model.py b/cognee/tests/unit/integration/test_model_to_graph_to_model.py
deleted file mode 100644
index 308c7ff15..000000000
--- a/cognee/tests/unit/integration/test_model_to_graph_to_model.py
+++ /dev/null
@@ -1,171 +0,0 @@
-from datetime import datetime, timezone
-from enum import Enum
-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"},
-    },
-)
-
-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",
-    },
-}
-
-PARSED_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"
-    Sedan = "Sedan"
-    SUV = "SUV"
-    Coupe = "Coupe"
-    Convertible = "Convertible"
-    Hatchback = "Hatchback"
-    Wagon = "Wagon"
-    Minivan = "Minivan"
-    Van = "Van"
-
-
-class CarType(DataPoint):
-    id: str
-    name: CarTypeName
-    _metadata: dict = dict(index_fields=["name"])
-
-
-class Car(DataPoint):
-    id: str
-    brand: str
-    model: str
-    year: int
-    color: str
-    is_type: CarType
-
-
-class Person(DataPoint):
-    id: str
-    name: str
-    age: int
-    owns_car: list[Car]
-    driving_license: Optional[dict]
-    _metadata: dict = dict(index_fields=["name"])
-
-
-def run_test_against_ground_truth(
-    test_target_item_name, test_target_item, ground_truth_dict
-):
-    for key, ground_truth in ground_truth_dict.items():
-        if isinstance(ground_truth, dict):
-            for key2, ground_truth2 in ground_truth.items():
-                assert (
-                    ground_truth2 == getattr(test_target_item, key)[key2]
-                ), f"{test_target_item_name}/{key = }/{key2 = }: {ground_truth2 = } != {getattr(test_target_item, key)[key2] = }"
-        else:
-            assert ground_truth == getattr(
-                test_target_item, key
-            ), f"{test_target_item_name}/{key = }: {ground_truth = } != {getattr(test_target_item, key) = }"
-    time_delta = datetime.now(timezone.utc) - getattr(test_target_item, "updated_at")
-
-    assert time_delta.total_seconds() < 20, f"{ time_delta.total_seconds() = }"
-
-
-@pytest.fixture(scope="session")
-def graph_outputs():
-    boris = Person(
-        id="boris",
-        name="Boris",
-        age=30,
-        owns_car=[
-            Car(
-                id="car1",
-                brand="Toyota",
-                model="Camry",
-                year=2020,
-                color="Blue",
-                is_type=CarType(id="sedan", name=CarTypeName.Sedan),
-            )
-        ],
-        driving_license={
-            "issued_by": "PU Vrsac",
-            "issued_on": "2025-11-06",
-            "number": "1234567890",
-            "expires_on": "2025-11-06",
-        },
-    )
-    nodes, edges = get_graph_from_model(boris)
-
-    car, person = nodes[0], nodes[1]
-    edge = edges[0]
-
-    parsed_person = get_model_instance_from_graph(nodes, edges, "boris")
-
-    return (car, person, edge, parsed_person)
-
-
-def test_extracted_person(graph_outputs):
-    (_, person, _, _) = graph_outputs
-
-    run_test_against_ground_truth("person", person, PERSON_GROUND_TRUTH)
-
-
-def test_extracted_car(graph_outputs):
-    (car, _, _, _) = graph_outputs
-    run_test_against_ground_truth("car", car, CAR_GROUND_TRUTH)
-
-
-def test_extracted_edge(graph_outputs):
-    (_, _, edge, _) = graph_outputs
-
-    assert (
-        EDGE_GROUND_TRUTH[:3] == edge[:3]
-    ), f"{EDGE_GROUND_TRUTH[:3] = } != {edge[:3] = }"
-    for key, ground_truth in EDGE_GROUND_TRUTH[3].items():
-        assert ground_truth == edge[3][key], f"{ground_truth = } != {edge[3][key] = }"
-
-
-def test_parsed_person(graph_outputs):
-    (_, _, _, parsed_person) = graph_outputs
-    run_test_against_ground_truth(
-        "parsed_person", parsed_person, PARSED_PERSON_GROUND_TRUTH
-    )
-    run_test_against_ground_truth("car", parsed_person.owns_car[0], CAR_GROUND_TRUTH)