add tests

Author: marvin-j97

src/table/filter/mod.rs

@@ -84,3 +84,28 @@ impl From<FilterType> for u8 {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use test_log::test;
+
+    #[test]
+    fn bloom_estimated_size_bpk() {
+        let policy = BloomConstructionPolicy::BitsPerKey(10.0);
+        let n = 1_000_000;
+        let estimated_size = policy.estimated_filter_size(n);
+        // For 1 million keys and 10 bits per key, the size should be around 1.25 MB
+        assert_eq!(estimated_size, 1_250_000);
+    }
+
+    #[test]
+    fn bloom_estimated_size_fpr() {
+        let policy = BloomConstructionPolicy::FalsePositiveRate(0.01);
+        let n = 1_000_000;
+        let estimated_size = policy.estimated_filter_size(n);
+        // For 1 million keys and 1% false positive rate, the size should be around 1.2 MB
+        assert!((estimated_size as f32) < 1_300_000.0);
+        assert!((estimated_size as f32) > 1_100_000.0);
+    }
+}