refactoring tests

prime-slam · Feb 25, 2024 · 5656bdd · 5656bdd
1 parent a8eb48a
commit 5656bdd
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 8 deletions.
diff --git a/tests/distance/test_connected_component.py b/tests/distance/test_connected_component.py
@@ -16,7 +16,9 @@
 import pytest
 import open3d as o3d
 
-from src.services.distance.connected_component import ExtractionLargestConnectedComponentProcessor
+from src.services.distance.connected_component import (
+    ExtractionLargestConnectedComponentProcessor,
+)
 
 
 @pytest.mark.parametrize(
@@ -98,8 +100,8 @@ def test_extract_largest_connected_component(
     expected_points,
     expected_trace,
 ):
-    actual_dist, actual_points, actual_trace = ExtractionLargestConnectedComponentProcessor().process(
-        dist, points, trace
+    actual_dist, actual_points, actual_trace = (
+        ExtractionLargestConnectedComponentProcessor().process(dist, points, trace)
     )
 
     assert (actual_dist == expected_dist).all()

diff --git a/tests/distance/test_isolated.py b/tests/distance/test_isolated.py
@@ -95,8 +95,8 @@ def test_remove_isolated_points(
     expected_points,
     expected_trace,
 ):
-    actual_dist, actual_points, actual_trace = RemovingIsolatedPointsProcessor().process(
-        dist, points, trace
+    actual_dist, actual_points, actual_trace = (
+        RemovingIsolatedPointsProcessor().process(dist, points, trace)
     )
 
     assert (actual_dist == expected_dist).all()

diff --git a/tests/utils/test_distance_matrix.py b/tests/utils/test_distance_matrix.py
@@ -121,14 +121,28 @@ def test_dfs(distance_matrix):
     actual_visited_vertices_1 = dfs(distance_matrix, start_vertex=1)
     expected_visited_vertices_1 = np.array(
         [
-            True, True, True, False, False, False, False, False,
+            True,
+            True,
+            True,
+            False,
+            False,
+            False,
+            False,
+            False,
         ]
     )
 
-    actual_visited_vertices_7 = dfs(distance_matrix, start_vertex=4)
+    actual_visited_vertices_7 = dfs(distance_matrix, start_vertex=7)
     expected_visited_vertices_7 = np.array(
         [
-            False, False, False, True, True, True, True, True,
+            False,
+            False,
+            False,
+            True,
+            True,
+            True,
+            True,
+            True,
         ]
     )
     assert (actual_visited_vertices_1 == expected_visited_vertices_1).all()