1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
extern crate roaring;
use proptest::collection::btree_set;
use proptest::prelude::*;
use roaring::RoaringBitmap;
#[test]
fn select() {
let bitmap = (0..2000).collect::<RoaringBitmap>();
assert_eq!(bitmap.select(0), Some(0));
}
#[test]
fn select_array() {
let bitmap = (0..2000).collect::<RoaringBitmap>();
assert_eq!(bitmap.select(0), Some(0));
assert_eq!(bitmap.select(100), Some(100));
assert_eq!(bitmap.select(1000), Some(1000));
assert_eq!(bitmap.select(1999), Some(1999));
assert_eq!(bitmap.select(2000), None);
}
#[test]
fn select_bitmap() {
let bitmap = (0..100_000).collect::<RoaringBitmap>();
assert_eq!(bitmap.select(0), Some(0));
assert_eq!(bitmap.select(63), Some(63));
assert_eq!(bitmap.select(1000), Some(1000));
assert_eq!(bitmap.select(65535), Some(65535));
}
#[test]
fn select_empty() {
let bitmap = RoaringBitmap::new();
assert_eq!(bitmap.select(0), None);
assert_eq!(bitmap.select(1024), None);
assert_eq!(bitmap.select(u32::MAX), None);
}
proptest! {
#[test]
fn proptest_select(values in btree_set(any::<u32>(), 1000)) {
let bitmap = RoaringBitmap::from_sorted_iter(values.iter().cloned()).unwrap();
for (i, value) in values.iter().cloned().enumerate() {
prop_assert_eq!(bitmap.select(i as u32), Some(value));
}
}
}
|
