add some tests

CoCo-Japan-pan · Jan 12, 2025 · 27d3ad7 · 27d3ad7
1 parent bf881ff
commit 27d3ad7
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/crates/misc/doubling/src/lib.rs b/crates/misc/doubling/src/lib.rs
@@ -63,4 +63,24 @@ mod test {
             assert_eq!(doubling.query(index, x as u64), expected);
         }
     }
+
+    #[test]
+    fn test_two_beki() {
+        let mut rng = thread_rng();
+        const SIZE: usize = 1024;
+        let func = (0..SIZE)
+            .map(|_| rng.gen_range(0..SIZE))
+            .collect::<Vec<_>>();
+        let max_pow = 1024;
+        let doubling = Doubling::new(&func, max_pow);
+        for bit in 0..=10 {
+            let index = rng.gen_range(0..SIZE);
+            let x = 1 << bit;
+            let expected = (0..x).fold(index, |i, _| func[i]);
+            assert_eq!(doubling.query(index, x as u64), expected);
+            let x = x - 1;
+            let expected = (0..x).fold(index, |i, _| func[i]);
+            assert_eq!(doubling.query(index, x as u64), expected);
+        }
+    }
 }