expand retain tests

The tests for `retain` previously examined the element. This adds a test examining the multiplicity as well.
lonnen · Nov 13, 2020 · e61c624 · e61c624
1 parent 635ac5e
commit e61c624
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test-lib.rs b/tests/test-lib.rs
@@ -408,4 +408,11 @@ fn test_retain() {
 
     assert_eq!(mset.get(&1), Some(&2usize));
     assert_eq!(mset.get(&2), Some(&2usize));
+
+    mset.retain(|&_k, v| {
+        v == 1
+    });
+
+    assert_eq!(mset.elements().len(), 1);
+    assert_eq!(mset.get(&2), Some(&1usize));
 }