Remove unused imports and make PdfDocument_test a pytest function

cognee/tests/unit/documents/PdfDocument_test.py

@@ -9,7 +9,7 @@ GROUND_TRUTH = [
 ]
 
 
-if __name__ == "__main__":
+def test_PdfDocument():
     test_file_path = os.path.join(
         os.sep,
         *(os.path.dirname(__file__).split(os.sep)[:-2]),

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

@@ -1,8 +1,3 @@
-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
 
 EDGE_GROUND_TRUTH = (

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

@@ -1,12 +1,9 @@
 from datetime import datetime, timezone
+from typing import Any, Dict
 
 
-from typing import Dict, Any
-
 def run_test_against_ground_truth(
-    test_target_item_name: str,
+    test_target_item_name: str, test_target_item: Any, ground_truth_dict: Dict[str, Any]
-    test_target_item: Any,
-    ground_truth_dict: Dict[str, Any]
 ):
     """Validates test target item attributes against ground truth values.