summaryrefslogtreecommitdiffstats
path: root/vendor/itertools/tests/peeking_take_while.rs
blob: 5be97271dd80ef0272b673433e33065461113554 (plain)
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
62
63
64
65
66
67
68
69
use itertools::Itertools;
use itertools::{put_back, put_back_n};

#[test]
fn peeking_take_while_peekable() {
    let mut r = (0..10).peekable();
    r.peeking_take_while(|x| *x <= 3).count();
    assert_eq!(r.next(), Some(4));
}

#[test]
fn peeking_take_while_put_back() {
    let mut r = put_back(0..10);
    r.peeking_take_while(|x| *x <= 3).count();
    assert_eq!(r.next(), Some(4));
    r.peeking_take_while(|_| true).count();
    assert_eq!(r.next(), None);
}

#[test]
fn peeking_take_while_put_back_n() {
    let mut r = put_back_n(6..10);
    for elt in (0..6).rev() {
        r.put_back(elt);
    }
    r.peeking_take_while(|x| *x <= 3).count();
    assert_eq!(r.next(), Some(4));
    r.peeking_take_while(|_| true).count();
    assert_eq!(r.next(), None);
}

#[test]
fn peeking_take_while_slice_iter() {
    let v = [1, 2, 3, 4, 5, 6];
    let mut r = v.iter();
    r.peeking_take_while(|x| **x <= 3).count();
    assert_eq!(r.next(), Some(&4));
    r.peeking_take_while(|_| true).count();
    assert_eq!(r.next(), None);
}

#[test]
fn peeking_take_while_slice_iter_rev() {
    let v = [1, 2, 3, 4, 5, 6];
    let mut r = v.iter().rev();
    r.peeking_take_while(|x| **x >= 3).count();
    assert_eq!(r.next(), Some(&2));
    r.peeking_take_while(|_| true).count();
    assert_eq!(r.next(), None);
}

#[test]
fn peeking_take_while_nested() {
    let mut xs = (0..10).peekable();
    let ys: Vec<_> = xs
        .peeking_take_while(|x| *x < 6)
        .peeking_take_while(|x| *x != 3)
        .collect();
    assert_eq!(ys, vec![0, 1, 2]);
    assert_eq!(xs.next(), Some(3));

    let mut xs = (4..10).peekable();
    let ys: Vec<_> = xs
        .peeking_take_while(|x| *x != 3)
        .peeking_take_while(|x| *x < 6)
        .collect();
    assert_eq!(ys, vec![4, 5]);
    assert_eq!(xs.next(), Some(6));
}