sparse_complex 0.1.4

A simple solver for complex sparse matrices.
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138

#[cfg(test)]
mod tests {
    use approx::assert_abs_diff_eq;
    use sparse_complex::*;
    use num::Complex;
    #[test]
    //#[ignore = "Solve not implemented"]
    fn test_simple_matrix() {
        let mut m = ComplexMatrix::<f64>::new();
        let mut b = vec![Complex::new(1., 0.), Complex::new(0., 1.)];
        m.add_element(0, 0, Complex::new(1., 1.));
        m.add_element(1, 1, Complex::new(1., 1.));
        m.solve(&mut b).unwrap();

        let expected = vec![Complex::new(0.5, -0.5), Complex::new(0.5, 0.5)];
        for (i, e) in b.iter().enumerate() {
            assert_abs_diff_eq!(e.re, &expected[i].re, epsilon = 1e-6);
            assert_abs_diff_eq!(e.im, &expected[i].im, epsilon = 1e-6);
        }
    }

    #[test]
    // #[ignore = "Solve not implemented"]
    fn test_full_simple_matrix() {
        let mut m = ComplexMatrix::<f64>::new();
        m.add_element(0, 0, Complex::new(5., 3.));
        m.add_element(1, 1, Complex::new(1., -9.));
        m.add_element(0, 1, Complex::new(-33., 0.));
        m.add_element(1, 0, Complex::new(0., -47.));

        let mut b = vec![Complex::new(13.4, 7.), Complex::new(3.2, -7.)];
        let expected = vec![
            Complex::new(0.21852826260018543, 0.10986007256547237),
            Complex::new(-0.3829375425665308, -0.1756095408900631),
        ];

        m.solve(&mut b).unwrap();

        for (i, e) in b.iter().enumerate() {
            assert_abs_diff_eq!(e.re, &expected[i].re, epsilon = 1e-6);
            assert_abs_diff_eq!(e.im, &expected[i].im, epsilon = 1e-6);
        }
    }

    #[test]
    fn test_only_imag_simple_matrix() {
        let mut m = ComplexMatrix::<f64>::new();
        m.add_element(0, 0, Complex::new(0., 3.));
        m.add_element(0, 1, Complex::new(0., -33.));
        m.add_element(1, 0, Complex::new(0., -1.));
        m.add_element(1, 1, Complex::new(0., 9.));

        let mut b = vec![Complex::new(0., 3.), Complex::new(0., 6.)];

        m.solve(&mut b).unwrap();

        let expected = vec![Complex::new(-37.5, 0.), Complex::new(-3.5, 0.)];
        for (i, e) in b.iter().enumerate() {
            assert_abs_diff_eq!(e.re, &expected[i].re, epsilon = 1e-6);
            assert_abs_diff_eq!(e.im, &expected[i].im, epsilon = 1e-6);
        }
    }

    #[test]
    fn test_only_real_simple_matrix() {
        let mut m = ComplexMatrix::<f64>::new();
        m.add_element(0, 0, Complex::new(3., 0.));
        m.add_element(0, 1, Complex::new(-7., 0.));
        m.add_element(1, 0, Complex::new(-1., 0.));
        m.add_element(1, 1, Complex::new(9., 0.));

        let mut b = vec![Complex::new(3., 0.), Complex::new(6., 0.)];

        m.solve(&mut b).unwrap();

        let expected = vec![Complex::new(3.4500000000000006, 0.), Complex::new(1.05, 0.)];
        for (i, e) in b.iter().enumerate() {
            assert_abs_diff_eq!(e.re, &expected[i].re, epsilon = 1e-6);
            assert_abs_diff_eq!(e.im, &expected[i].im, epsilon = 1e-6);
        }
    }

    #[test]
    fn test_part_real_part_imag_simple_matrix() {
        let mut m = ComplexMatrix::<f64>::new();
        m.add_element(0, 0, Complex::new(3., 0.));
        m.add_element(0, 1, Complex::new(0., 1.));
        m.add_element(1, 0, Complex::new(0., -4.));
        m.add_element(1, 1, Complex::new(1., 0.));

        let mut b = vec![Complex::new(0., 3.), Complex::new(6., 0.)];

        m.solve(&mut b).unwrap();

        let expected = vec![Complex::new(0., 3.), Complex::new(-6., 0.)];
        for (i, e) in b.iter().enumerate() {
            assert_abs_diff_eq!(e.re, &expected[i].re, epsilon = 1e-6);
            assert_abs_diff_eq!(e.im, &expected[i].im, epsilon = 1e-6);
        }
    }

    #[test]
    fn test_b_imag_is_zero() {
        let mut m = ComplexMatrix::<f64>::new();
        m.add_element(0, 0, Complex::new(3., 0.));
        m.add_element(0, 1, Complex::new(0., 1.));
        m.add_element(1, 0, Complex::new(0., -4.));
        m.add_element(1, 1, Complex::new(1., 0.));

        let mut b = vec![Complex::new(1., 0.), Complex::new(5., 0.)];

        m.solve(&mut b).unwrap();
        let expected = vec![Complex::new(-1., 5.), Complex::new(-15., -4.)];
        for (i, e) in b.iter().enumerate() {
            assert_abs_diff_eq!(e.re, &expected[i].re, epsilon = 1e-6);
            assert_abs_diff_eq!(e.im, &expected[i].im, epsilon = 1e-6);
        }
    }

    #[test]
    fn test_b_real_is_zero() {
        let mut m = ComplexMatrix::<f64>::new();
        m.add_element(0, 0, Complex::new(3., 0.));
        m.add_element(0, 1, Complex::new(0., 1.));
        m.add_element(1, 0, Complex::new(0., -4.));
        m.add_element(1, 1, Complex::new(1., 0.));

        let mut b = vec![Complex::new(0., 1.), Complex::new(0., 5.)];

        m.solve(&mut b).unwrap();

        let expected = vec![Complex::new(-5., -1.), Complex::new(4., -15.)];
        for (i, e) in b.iter().enumerate() {
            assert_abs_diff_eq!(e.re, &expected[i].re, epsilon = 1e-6);
            assert_abs_diff_eq!(e.im, &expected[i].im, epsilon = 1e-6);
        }
    }
}