@@ -187,11 +187,11 @@ def _test_minimum_spanning_tree(func, ds, algorithm, *args):
187
187
188
188
def _test_minimum_spanning_tree_cpp (ds , algorithm , * args ):
189
189
if (ds == 'List' and algorithm == "prim" ):
190
- a1 = AdjacencyListGraphNode ('a' ,0 , backend = Backend .CPP )
191
- b1 = AdjacencyListGraphNode ('b' ,0 , backend = Backend .CPP )
192
- c1 = AdjacencyListGraphNode ('c' ,0 , backend = Backend .CPP )
193
- d1 = AdjacencyListGraphNode ('d' ,0 , backend = Backend .CPP )
194
- e1 = AdjacencyListGraphNode ('e' ,0 , backend = Backend .CPP )
190
+ a1 = AdjacencyListGraphNode ('a' , 0 , backend = Backend .CPP )
191
+ b1 = AdjacencyListGraphNode ('b' , 0 , backend = Backend .CPP )
192
+ c1 = AdjacencyListGraphNode ('c' , 0 , backend = Backend .CPP )
193
+ d1 = AdjacencyListGraphNode ('d' , 0 , backend = Backend .CPP )
194
+ e1 = AdjacencyListGraphNode ('e' , 0 , backend = Backend .CPP )
195
195
g = Graph (a1 , b1 , c1 , d1 , e1 , backend = Backend .CPP )
196
196
g .add_edge (a1 .name , c1 .name , 10 )
197
197
g .add_edge (c1 .name , a1 .name , 10 )
@@ -206,10 +206,10 @@ def _test_minimum_spanning_tree_cpp(ds, algorithm, *args):
206
206
mst = minimum_spanning_tree (g , "prim" , backend = Backend .CPP )
207
207
expected_mst = ["('a', 'd', 7)" , "('d', 'c', 9)" , "('e', 'd', 23)" , "('b', 'd', 32)" ,
208
208
"('d', 'a', 7)" , "('c', 'd', 9)" , "('d', 'e', 23)" , "('d', 'b', 32)" ]
209
- assert str (mst .get_edge ('a' ,'d' )) in expected_mst
210
- assert str (mst .get_edge ('e' ,'d' )) in expected_mst
211
- assert str (mst .get_edge ('d' ,'c' )) in expected_mst
212
- assert str (mst .get_edge ('b' ,'d' )) in expected_mst
209
+ assert str (mst .get_edge ('a' , 'd' )) in expected_mst
210
+ assert str (mst .get_edge ('e' , 'd' )) in expected_mst
211
+ assert str (mst .get_edge ('d' , 'c' )) in expected_mst
212
+ assert str (mst .get_edge ('b' , 'd' )) in expected_mst
213
213
assert mst .num_edges () == 8
214
214
a = AdjacencyListGraphNode ('0' , 0 , backend = Backend .CPP )
215
215
b = AdjacencyListGraphNode ('1' , 0 , backend = Backend .CPP )
@@ -223,7 +223,7 @@ def _test_minimum_spanning_tree_cpp(ds, algorithm, *args):
223
223
g2 .add_edge ('1' , '2' , 74 )
224
224
g2 .add_edge ('2' , '1' , 74 )
225
225
mst2 = minimum_spanning_tree (g2 , "prim" , backend = Backend .CPP )
226
- assert mst2 .num_edges ()== 6
226
+ assert mst2 .num_edges () == 6
227
227
228
228
fmst = minimum_spanning_tree
229
229
fmstp = minimum_spanning_tree_parallel
0 commit comments