scirs2-linalg 0.4.0

Linear algebra module for SciRS2 (scirs2-linalg)
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

//! Tests for exponential matrix functions

use super::super::exponential::*;
use scirs2_core::ndarray::array;

#[test]
fn test_expm_identity() {
    let identity = array![[1.0_f64, 0.0], [0.0, 1.0]];
    let zero = array![[0.0_f64, 0.0], [0.0, 0.0]];
    let result = expm(&zero.view(), None).expect("Test: operation failed");

    for i in 0..2 {
        for j in 0..2 {
            assert!((result[[i, j]] - identity[[i, j]]).abs() < 1e-10);
        }
    }
}

#[test]
fn test_logm_identity() {
    let identity = array![[1.0_f64, 0.0], [0.0, 1.0]];
    let result = logm(&identity.view()).expect("Test: operation failed");

    for i in 0..2 {
        for j in 0..2 {
            assert!(result[[i, j]].abs() < 1e-10);
        }
    }
}

#[test]
fn test_sqrtm_diagonal() {
    let a = array![[4.0_f64, 0.0], [0.0, 9.0]];
    let result = sqrtm(&a.view(), 20, 1e-10).expect("Test: operation failed");

    assert!((result[[0, 0]] - 2.0).abs() < 1e-10);
    assert!((result[[0, 1]] - 0.0).abs() < 1e-10);
    assert!((result[[1, 0]] - 0.0).abs() < 1e-10);
    assert!((result[[1, 1]] - 3.0).abs() < 1e-10);
}

#[test]
fn test_matrix_power_integer() {
    let a = array![[2.0_f64, 0.0], [0.0, 3.0]];
    let result = matrix_power(&a.view(), 2.0).expect("Test: operation failed");

    assert!((result[[0, 0]] - 4.0).abs() < 1e-10);
    assert!((result[[0, 1]] - 0.0).abs() < 1e-10);
    assert!((result[[1, 0]] - 0.0).abs() < 1e-10);
    assert!((result[[1, 1]] - 9.0).abs() < 1e-10);
}