wicker 0.2.0

Weighted probability picker for Rust
Documentation 
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
53
54
55
56
57
58
59
60
61

use wicker::*;

#[test]
fn double_ended() {
    let picker = WeightedPicker::new(vec![
        ("hello", 0.3),
        ("wicker", 10.0),
        ("this", 3.0),
        ("world", 1.0),
        ("is", 5.0),
    ]);
    eprintln!("{:?}", &picker);
    let mut iter = picker.iter().copied();
    assert_eq!(iter.next_back(), Some("wicker"));
    assert_eq!(iter.next_back(), Some("is"));
    assert_eq!(iter.next_back(), Some("this"));
    assert_eq!(iter.next_back(), Some("world"));
    assert_eq!(iter.next_back(), Some("hello"));
    assert_eq!(iter.next_back(), None);
}

#[test]
fn both_ends() {
    let picker = WeightedPicker::new(vec![
        ("hello", 0.3),
        ("wicker", 10.0),
        ("this", 3.0),
        ("world", 1.0),
        ("is", 5.0),
    ]);
    let mut iter = picker.iter().copied();
    assert_eq!(iter.next_back(), Some("wicker"));
    assert_eq!(iter.next(), Some("hello"));
    assert_eq!(iter.next(), Some("world"));
    assert_eq!(iter.next_back(), Some("is"));
    assert_eq!(iter.next_back(), Some("this"));
    assert_eq!(iter.next(), None);
    assert_eq!(iter.next_back(), None);
}

#[test]
fn exact_size() {
    let picker = WeightedPicker::new(vec![
        ("hello", 0.3),
        ("wicker", 10.0),
        ("this", 3.0),
        ("world", 1.0),
        ("is", 5.0),
    ]);
    let mut iter = picker.iter().copied();
    for idx in 0..=picker.len() {
        let expect = picker.len() - idx;
        assert_eq!(expect, iter.len());
        // get a mix of front and back
        if idx % 2 == 0 {
            iter.next();
        } else {
            iter.next_back();
        }
    }
}