arcium-primitives 0.4.2

Arcium primitives
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

pub fn transpose<Outer, Inner, T>(matrix: Outer) -> Vec<Vec<T>>
where
    Outer: IntoIterator<Item = Inner>,
    Inner: IntoIterator<Item = T>,
{
    let mut row_iters: Vec<_> = matrix.into_iter().map(|row| row.into_iter()).collect();

    if row_iters.is_empty() {
        return vec![];
    }

    let mut result = Vec::new();

    loop {
        let mut column: Vec<T> = Vec::new();
        let mut has_elements = false;

        for row_iter in row_iters.iter_mut() {
            if let Some(element) = row_iter.next() {
                column.push(element);
                has_elements = true;
            }
        }

        if !has_elements {
            break;
        }

        result.push(column);
    }

    result
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_transpose_empty_matrix() {
        let matrix: Vec<Vec<i32>> = vec![];
        let result = transpose(matrix.clone());
        assert_eq!(result, matrix);
    }

    #[test]
    fn test_transpose_empty_rows() {
        let matrix: Vec<Vec<i32>> = vec![];
        let result = transpose(matrix.clone());
        assert_eq!(result, matrix);
    }

    #[test]
    fn test_transpose_single_element() {
        let matrix = vec![vec![1]];
        let result = transpose(matrix);
        assert_eq!(result, vec![vec![1]]);
    }

    #[test]
    fn test_transpose_single_row() {
        let matrix = vec![vec![1, 2, 3]];
        let result = transpose(matrix);
        assert_eq!(result, vec![vec![1], vec![2], vec![3]]);
    }

    #[test]
    fn test_transpose_single_column() {
        let matrix = vec![vec![1], vec![2], vec![3]];
        let result = transpose(matrix);
        assert_eq!(result, vec![vec![1, 2, 3]]);
    }

    #[test]
    fn test_transpose_square_matrix() {
        let matrix = vec![vec![1, 2, 3], vec![4, 5, 6], vec![7, 8, 9]];
        let result = transpose(matrix);
        let expected = vec![vec![1, 4, 7], vec![2, 5, 8], vec![3, 6, 9]];
        assert_eq!(result, expected);
    }

    #[test]
    fn test_transpose_rectangular_matrix() {
        let matrix = vec![vec![1, 2, 3, 4], vec![5, 6, 7, 8]];
        let result = transpose(matrix);
        let expected = vec![vec![1, 5], vec![2, 6], vec![3, 7], vec![4, 8]];
        assert_eq!(result, expected);
    }

    #[test]
    fn test_transpose_with_strings() {
        let matrix = vec![vec!["a", "b"], vec!["c", "d"], vec!["e", "f"]];
        let result = transpose(matrix);
        let expected = vec![vec!["a", "c", "e"], vec!["b", "d", "f"]];
        assert_eq!(result, expected);
    }

    #[test]
    fn test_transpose_double_transpose() {
        let matrix = vec![vec![1, 2, 3], vec![4, 5, 6]];
        let result = transpose(transpose(matrix.clone()));
        assert_eq!(result, matrix);
    }
}