Test MST's cost

icode-live · Jun 9, 2017 · 4a45615 · 4a45615
1 parent c9511c6
commit 4a45615
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/graph/mod.rs b/src/graph/mod.rs
@@ -158,6 +158,9 @@ mod test {
         graph.add_undirected_edge(1, 2);
         graph.add_undirected_edge(2, 0);
         let weights = [7, 3, 5];
-        assert_eq!(graph.min_spanning_tree(&weights), vec![1, 2]);
+        let mst = graph.min_spanning_tree(&weights);
+        let mst_cost = mst.iter().map(|&e| weights[e]).sum::<i64>();
+        assert_eq!(mst, vec![1, 2]);
+        assert_eq!(mst_cost, 8);
     }
 }