From 24b9eaf923d05191b62448dbe4e5e3217255a62d Mon Sep 17 00:00:00 2001
From: Bhavesh9908 <madhavibhavesh558@gmail.com>
Date: Tue, 3 Sep 2024 20:04:19 +0530
Subject: [PATCH] Bhavesh9908

---
 README.md                                    |  5 ++-
 data_structures/kd_tree/tests/test_kdtree.py | 37 ++++++--------------
 2 files changed, 15 insertions(+), 27 deletions(-)

diff --git a/README.md b/README.md
index d8eba4e016fa..fde39af7d6aa 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,5 @@
 <div align="center">
 <!-- Title: -->
-  <a href="https://github.com/TheAlgorithms/">
     <img src="https://raw.githubusercontent.com/TheAlgorithms/website/1cd824df116b27029f17c2d1b42d81731f28a920/public/logo.svg" height="100">
   </a>
   <h1><a href="https://github.com/TheAlgorithms/">The Algorithms</a> - Python</h1>
@@ -30,6 +29,10 @@
   <a href="https://github.com/psf/black">
     <img src="https://img.shields.io/static/v1?label=code%20style&message=black&color=black&style=flat-square" height="20" alt="code style: black">
   </a>
+  </a>
+  <a href="https://github.com/Bhavesh9908">
+    <img src="https://avatars.githubusercontent.com/u/141760947?v=4 alt="pre-commit>
+  </a>
 <!-- Short description: -->
   <h3>All algorithms implemented in Python - for education</h3>
 </div>
diff --git a/data_structures/kd_tree/tests/test_kdtree.py b/data_structures/kd_tree/tests/test_kdtree.py
index 81f2cc990074..d04648e46a8e 100644
--- a/data_structures/kd_tree/tests/test_kdtree.py
+++ b/data_structures/kd_tree/tests/test_kdtree.py
@@ -8,46 +8,33 @@
 
 
 @pytest.mark.parametrize(
-    ("num_points", "cube_size", "num_dimensions", "depth", "expected_result"),
+    ("num_points", "cube_size", "num_dimensions", "expected_result"),
     [
-        (0, 10.0, 2, 0, None),  # Empty points list
-        (10, 10.0, 2, 2, KDNode),  # Depth = 2, 2D points
-        (10, 10.0, 3, -2, KDNode),  # Depth = -2, 3D points
+        (0, 10.0, 2, None),  # Empty points list
+        (10, 10.0, 2, KDNode),  # 2D points
+        (10, 10.0, 3, KDNode),  # 3D points
     ],
 )
-def test_build_kdtree(num_points, cube_size, num_dimensions, depth, expected_result):
+def test_build_kdtree(num_points, cube_size, num_dimensions, expected_result):
     """
     Test that KD-Tree is built correctly.
 
     Cases:
         - Empty points list.
-        - Positive depth value.
-        - Negative depth value.
+        - Valid points list with correct dimensions.
     """
-    points = (
-        hypercube_points(num_points, cube_size, num_dimensions).tolist()
-        if num_points > 0
-        else []
-    )
+    points = hypercube_points(num_points, cube_size, num_dimensions).tolist() if num_points > 0 else []
 
-    kdtree = build_kdtree(points, depth=depth)
+    kdtree = build_kdtree(points)
 
     if expected_result is None:
         # Empty points list case
         assert kdtree is None, f"Expected None for empty points list, got {kdtree}"
     else:
-        # Check if root node is not None
+        # Check if KD-Tree is built correctly
         assert kdtree is not None, "Expected a KDNode, got None"
-
-        # Check if root has correct dimensions
-        assert (
-            len(kdtree.point) == num_dimensions
-        ), f"Expected point dimension {num_dimensions}, got {len(kdtree.point)}"
-
-        # Check that the tree is balanced to some extent (simplistic check)
-        assert isinstance(
-            kdtree, KDNode
-        ), f"Expected KDNode instance, got {type(kdtree)}"
+        assert isinstance(kdtree, KDNode), f"Expected KDNode instance, got {type(kdtree)}"
+        assert len(kdtree.point) == num_dimensions, f"Expected point dimension {num_dimensions}, got {len(kdtree.point)}"
 
 
 def test_nearest_neighbour_search():
@@ -95,6 +82,4 @@ def test_edge_cases():
 
 
 if __name__ == "__main__":
-    import pytest
-
     pytest.main()