bugfix: Use 3D position keys in edge creation test

funkelab · Feb 28, 2024 · 7668474 · 7668474
1 parent 0f6f529
commit 7668474
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_candidate_graph/test_graph_from_segmentation.py b/tests/test_candidate_graph/test_graph_from_segmentation.py
@@ -157,7 +157,7 @@ def test_add_cand_edges_2d(graph_2d):
 
 def test_add_cand_edges_3d(graph_3d):
     cand_graph = nx.create_empty_copy(graph_3d)
-    add_cand_edges(cand_graph, max_edge_distance=15)
+    add_cand_edges(cand_graph, max_edge_distance=15, position_keys=("z", "y", "x"))
     graph_3d.remove_edge("0_1", "1_1")
     assert Counter(list(cand_graph.edges)) == Counter(list(graph_3d.edges))
     for edge in cand_graph.edges: