Try to find empty subbitmaps in treemap

RoaringBitmap · Jan 28, 2024 · a236172 · a236172
1 parent 1dae104
commit a236172
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/croaring/src/treemap/iter.rs b/croaring/src/treemap/iter.rs
@@ -39,7 +39,11 @@ pub struct TreemapIterator<'a> {
 
 impl<'a> TreemapIterator<'a> {
     fn new(treemap: &'a Treemap) -> Self {
-        let iter = treemap.map.iter().flat_map(to64iter as _);
+        let iter = treemap
+            .map
+            .iter()
+            .inspect(|(key, bitmap)| assert!(!bitmap.is_empty(), "empty bitmap at {key}"))
+            .flat_map(to64iter as _);
 
         TreemapIterator { iter }
     }

diff --git a/fuzz/fuzz_targets/arbitrary_ops64/mod.rs b/fuzz/fuzz_targets/arbitrary_ops64/mod.rs
@@ -417,7 +417,7 @@ pub fn assert_64_eq(lhs: &Bitmap64, rhs: &Treemap) {
             let r = rhs.next().unwrap();
             assert_eq!(l, r, "{l} != {r} at {i}");
         }
-        assert!(rhs.next().is_none());
+        assert_eq!(rhs.next(), None);
         panic!("Serialize not equal, but all items equal?")
     }
 }