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
|
use std::str::FromStr;
use crate::{error::ParseMatrixError, matrix::Matrix, MatrixMath};
enum TestCaseType {
Add,
Mul,
Inv,
CmpErr,
}
struct TestCase {
test_type: TestCaseType,
test_data: Vec<Matrix>,
}
fn build_add_test_cases() -> Vec<TestCase> {
let mut v = vec![];
let from_strs = vec![
"1,2,3\n4,5,6\n7,8,9",
"1,1,1\n1,1,1\n1,1,1",
"2,3,4\n5,6,7\n8,9,10",
"1,1,1\n1,1,1\n1,1,1",
"0,0,0\n0,0,0\n0,0,0",
"1,1,1\n1,1,1\n1,1,1",
];
let mut i = 0;
while i < from_strs.len() {
let m1 = Matrix::from_str(from_strs[i]).unwrap();
let m2 = Matrix::from_str(from_strs[i+1]).unwrap();
let mr = Matrix::from_str(from_strs[i+2]).unwrap();
v.push(TestCase {
test_type: TestCaseType::Add,
test_data: vec![m1, m2, mr],
});
i += 3;
}
v
}
#[test]
pub fn test_matrix_add() -> Result<(), ParseMatrixError> {
let cases = build_add_test_cases();
for case in cases {
assert_eq!(&case.test_data[0] + &case.test_data[1], case.test_data[2]);
}
Ok(())
}
#[test]
pub fn test_matrix_determinate() -> Result<(), ParseMatrixError> {
let m = Matrix::from_str("3,4\n5,6")?;
let det = 3.0 * 6.0 - 4.0 * 5.0;
assert_eq!(m.determinant(), det);
Ok(())
}
#[test]
pub fn test_matrix_inverse_on_singular() -> Result<(), ()> {
let m = Matrix::new(vec![vec![1.0,2.0,3.0], vec![4.0,5.0,6.0], vec![7.0,8.0,9.0]]);
match m.inverse() {
Some(_inverse) => Err(()),
None => Ok(()),
}
}
#[test]
pub fn test_matrix_transpose() -> Result<(), ParseMatrixError> {
let m = Matrix::from_str("1,2,3\n4,5,6\n7,8,9")?;
let t = Matrix::from_str("1,4,7\n2,5,8\n3,6,9")?;
assert_eq!(m.transpose(), t);
Ok(())
}
#[test]
pub fn test_matrix_mul() -> Result<(), ParseMatrixError> {
let m1 = Matrix::from_str("1,2\n1,2")?;
let m2 = Matrix::from_str("1,3\n2,4")?;
let m3 = Matrix::from_str("1,2\n3,4")?;
let m4 = Matrix::from_str("1\n2")?;
let t1 = Matrix::from_str("5,11\n5,11")?;
let t2 = Matrix::from_str("5\n11")?;
assert_eq!(&m1 * &m2, t1);
assert_eq!(&m3 * &m4, t2);
Ok(())
}
#[test]
#[should_panic]
pub fn test_matrix_add_bad_dimensions() -> () {
let bad = Matrix::from_str("1,1,1\n1,1,1").unwrap();
let add = Matrix::from_str("1,1\n1,1").unwrap();
let _ = &bad + &add;
}
|
