Test shift_insert that moves

indexmap-rs · Feb 11, 2024 · 8c206ef · 8c206ef
1 parent 5debe73
commit 8c206ef
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/map/tests.rs b/src/map/tests.rs
@@ -125,6 +125,14 @@ fn shift_insert() {
     for (i, k) in (0..insert.len()).zip(map.keys()) {
         assert_eq!(map.get_index(i).unwrap().0, k);
     }
+
+    // "insert" that moves an existing entry
+    map.shift_insert(0, insert[0], ());
+    assert_eq!(map.keys().count(), insert.len());
+    assert_eq!(insert[0], map.keys()[0]);
+    for (a, b) in insert[1..].iter().rev().zip(map.keys().skip(1)) {
+        assert_eq!(a, b);
+    }
 }
 
 #[test]

diff --git a/src/set/tests.rs b/src/set/tests.rs
@@ -144,6 +144,14 @@ fn shift_insert() {
     for (i, v) in (0..insert.len()).zip(set.iter()) {
         assert_eq!(set.get_index(i).unwrap(), v);
     }
+
+    // "insert" that moves an existing entry
+    set.shift_insert(0, insert[0]);
+    assert_eq!(set.iter().count(), insert.len());
+    assert_eq!(insert[0], set[0]);
+    for (a, b) in insert[1..].iter().rev().zip(set.iter().skip(1)) {
+        assert_eq!(a, b);
+    }
 }
 
 #[test]