summaryrefslogtreecommitdiffstats
path: root/third_party/rust/rayon/tests/drain_vec.rs
blob: 08f1120b799aaf51d7ab23b58c58b387836f836c (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
use rayon::prelude::*;

#[test]
fn drain_vec_yielded() {
    let mut vec_org = vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9];

    let yielded = vec_org.par_drain(0..5).collect::<Vec<_>>();

    assert_eq!(&yielded, &[0, 1, 2, 3, 4]);
    assert_eq!(&vec_org, &[5, 6, 7, 8, 9]);
}

#[test]
fn drain_vec_dropped() {
    let mut vec_org = vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9];

    let yielded = vec_org.par_drain(0..5);

    drop(yielded);
    assert_eq!(&vec_org, &[5, 6, 7, 8, 9]);
}

#[test]
fn drain_vec_empty_range_yielded() {
    let mut vec_org = vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9];

    let yielded = vec_org.par_drain(5..5).collect::<Vec<_>>();

    assert_eq!(&yielded, &[]);
    assert_eq!(&vec_org, &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]);
}

#[test]
fn drain_vec_empty_range_dropped() {
    let mut vec_org = vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9];

    let yielded = vec_org.par_drain(5..5);

    drop(yielded);
    assert_eq!(&vec_org, &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]);
}