mermaid 0.1.0

Experimental Linear Algebra
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

use super::Matrix;
use std::ops::{Add, AddAssign};

impl<T, U, const M: usize, const N: usize> Add for Matrix<T, M, N>
where
    T: Add<Output = U>,
{
    type Output = Matrix<U, M, N>;

    fn add(self, rhs: Self) -> Self::Output {
        Self::Output::new(
            self.0
                .into_iter()
                .zip(rhs.0.into_iter())
                .map(|(a, b)| a + b)
                .collect(),
        )
        .expect("entries is the same size as matrix")
    }
}

impl<T, const M: usize, const N: usize> AddAssign for Matrix<T, M, N>
where
    T: AddAssign,
{
    fn add_assign(&mut self, rhs: Self) {
        self.0.iter_mut().zip(rhs.0.into_iter()).for_each(|(a, b)| {
            *a += b;
        });
    }
}

#[cfg(test)]
mod tests {
    use super::super::Matrix;
    use std::error::Error;

    #[test]
    fn addition() -> Result<(), Box<dyn Error>> {
        let m1: Matrix<_, 2, 5> = Matrix::new(vec![5u32; 10])?;
        let m2: Matrix<_, 2, 5> = Matrix::new(vec![2u32; 10])?;

        let expected = Matrix::new(vec![7u32; 10])?;
        let actual = m1 + m2;
        assert_eq!(expected, actual);

        let mut m1: Matrix<_, 3, 7> = Matrix::new(vec![9i32; 21])?;
        let m2: Matrix<_, 3, 7> = Matrix::new(vec![1i32; 21])?;

        let expected = Matrix::new(vec![10i32; 21])?;
        m1 += m2;
        assert_eq!(expected, m1);

        Ok(())
    }
}