nullvec 0.2.0

Rust nullable vector, which can contain null(missing) values as element
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


pub struct Indexing;

impl Indexing {
    pub fn blocs<T: Clone>(values: &[T], flags: &[bool]) -> Vec<T> {
        assert!(
            values.len() == flags.len(),
            "flags must be the same length as values"
        );
        // should use filter_map?
        let new_values: Vec<T> = values
            .iter()
            .zip(flags.iter())
            .filter(|&(_, y)| *y)
            .map(|(ref x, _)| (*x).clone())
            .collect();
        new_values
    }

    /// reorder values based on given locations
    pub fn reindex<T: Clone>(values: &[T], locs: &[usize]) -> Vec<T> {
        Self::assert_index_boundary(values, locs);
        unsafe { Self::reindex_unchecked(values, locs) }
    }

    /// reorder values based on given locations
    pub unsafe fn reindex_unchecked<T: Clone>(values: &[T], locs: &[usize]) -> Vec<T> {
        locs.iter()
            .map(|&i| values.get_unchecked(i).clone())
            .collect()
    }

    pub fn assert_index_boundary<T>(values: &[T], locs: &[usize]) {
        let len = values.len();
        assert!(locs.iter().all(|&i| i < len), "Index out of bounds");
    }
}

#[cfg(test)]
mod tests {

    use super::Indexing;

    #[test]
    fn test_blocs() {
        let i = vec![1, 2, 3];
        let f = vec![true, false, true];
        assert_eq!(Indexing::blocs(&i, &f), vec![1, 3]);

        let s: Vec<String> = vec!["a".to_string(), "b".to_string(), "c".to_string()];
        assert_eq!(
            Indexing::blocs(&s, &f),
            vec!["a".to_string(), "c".to_string()]
        );
    }

    #[test]
    #[should_panic]
    fn test_blocs_length_mismatch() {
        let i = vec![1, 2, 3, 4];
        let f = vec![true, false, true];
        Indexing::blocs(&i, &f);
    }

    #[test]
    fn test_reindex() {
        let res = Indexing::reindex(&vec![1, 2, 3, 4, 5], &vec![4, 2, 3]);
        assert_eq!(res, vec![5, 3, 4]);
    }
}