select_indices 3.0.0

Iterators for taking multiple shared/exclusive references from a slice
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use select_indices::prelude::*;
use ndarray::prelude::*;

#[test]
fn select_indices_test()
{
    let data = arr2(&[
        [0,0,2,0,0],
        [0,1,0,3,0],
        [0,0,4,5,0],
    ]);

    let indices = [
        (1, 1), (0, 2),
        (1, 3), (2, 2),
        (2, 3), (2, 3),
        (2, 2), (1, 3),
        (0, 2), (1, 1),
    ];

    assert!(
        data.select_indices(&indices).eq(&[1,2,3,4,5,5,4,3,2,1])
    )
}

#[test]
#[should_panic]
fn out_of_range_panic()
{
    let data = arr2(&[
        [1,0,0],
        [0,2,0],
        [0,0,3],
    ]);
    let indices = [
        [1,1],[2,2],[3,3]
    ];

    data.select_indices(&indices) // 3 is out of bounds: should panic
        .for_each(|x| {
            println!("{x}");
        })
}

#[test]
fn indexed()
{
    let data = arr2(&[
        [11, 22, 33, 44, 55, 66, 77, 88],
        [99, 00, 11, 22, 33, 44, 55, 66],
        [77, 88, 99, 00, 11, 22, 33, 44],
    ]);
    
    data.select_indices(&[(0,4), (2,7), (1,3), (0,0), (2,3)]).indexed().for_each(|(i, x)| {
        println!("data[{i:?}] = {x:02}");
    });
}


#[cfg(feature = "rayon")]
mod rayon {
    use select_indices::prelude::*;
    use ndarray::prelude::*;
    use rayon::prelude::*;

    #[test]
    fn select_indices_test()
    {
        let data = arr2(&[
            [0,0,2,0,0],
            [0,1,0,3,0],
            [0,0,4,5,0],
        ]);

        let indices = [
            (1, 1), (0, 2),
            (1, 3), (2, 2),
            (2, 3), (2, 3),
            (2, 2), (1, 3),
            (0, 2), (1, 1),
        ];

        assert!(
            data.par_select_indices(&indices).eq(&[1,2,3,4,5,5,4,3,2,1])
        )
    }

    #[test]
    #[should_panic]
    fn out_of_range_panic()
    {
        let data = arr2(&[
            [1,0,0],
            [0,2,0],
            [0,0,3],
        ]);
        let indices = [
            [1,1],[2,2],[3,3]
        ];

        data.par_select_indices(&indices) // 3 is out of bounds: should panic
            .for_each(|x| {
                println!("{x}");
            })
    }

    #[test]
    fn indexed()
    {
        let data = arr2(&[
            [11, 22, 33, 44, 55, 66, 77, 88],
            [99, 00, 11, 22, 33, 44, 55, 66],
            [77, 88, 99, 00, 11, 22, 33, 44],
        ]);
        
        data.par_select_indices(&[(0,4), (2,7), (1,3), (0,0), (2,3)]).indexed().for_each(|(i, x)| {
            println!("data[{i:?}] = {x:02}");
        });
    }
}