nice

Author: molpopgen

tests/test_edge_difference_iteration.rs

@@ -72,7 +72,20 @@ fn test_collect_edge_diff_iterators() {
     }
 
     let removals = diffs.iter().flat_map(|d| d.removals()).collect::<Vec<_>>();
-    let insertions = diffs.iter().map(|i| i.insertions()).collect::<Vec<_>>();
+    let insertions = diffs
+        .iter()
+        .flat_map(|d| d.insertions())
+        .collect::<Vec<_>>();
+    assert_eq!(insertions.len(), ts.edge_insertion_order().len());
+    for (i, j) in insertions.iter().zip(ts.edge_insertion_order().iter()) {
+        assert_eq!(
+            i.parent(),
+            ts.tables().edges().parent_column()[j.as_usize()]
+        );
+        assert_eq!(i.child(), ts.tables().edges().child_column()[j.as_usize()]);
+        assert_eq!(i.left(), ts.tables().edges().left_column()[j.as_usize()]);
+        assert_eq!(i.right(), ts.tables().edges().right_column()[j.as_usize()]);
+    }
 
     let removal_order = ts.edge_removal_order();
     // Removals have some nuance:
@@ -92,5 +105,8 @@ fn test_collect_edge_diff_iterators() {
             i.parent(),
             ts.tables().edges().parent_column()[j.as_usize()]
         );
+        assert_eq!(i.child(), ts.tables().edges().child_column()[j.as_usize()]);
+        assert_eq!(i.left(), ts.tables().edges().left_column()[j.as_usize()]);
+        assert_eq!(i.right(), ts.tables().edges().right_column()[j.as_usize()]);
     }
 }