2013-10-08 07:22:56 +08:00
|
|
|
use std::num::{Real, abs};
|
2013-06-28 00:16:07 +08:00
|
|
|
use std::rand::random;
|
2013-10-18 04:40:44 +08:00
|
|
|
use na::{Vec1, Vec3, Mat1, Mat2, Mat3, Mat4, Mat5, Mat6, Rot3, DMat, DVec, Indexable};
|
2013-10-08 07:22:56 +08:00
|
|
|
use na;
|
2013-05-19 01:04:03 +08:00
|
|
|
|
2013-05-22 07:15:03 +08:00
|
|
|
macro_rules! test_inv_mat_impl(
|
2013-06-09 22:04:54 +08:00
|
|
|
($t: ty) => (
|
2013-11-27 18:16:16 +08:00
|
|
|
10000.times(|| {
|
2013-05-22 07:15:03 +08:00
|
|
|
let randmat : $t = random();
|
|
|
|
|
2014-01-10 03:48:30 +08:00
|
|
|
assert!(na::approx_eq(&(na::inv(&randmat).unwrap() * randmat), &na::one()));
|
2013-11-27 18:16:16 +08:00
|
|
|
})
|
2013-05-22 07:15:03 +08:00
|
|
|
);
|
|
|
|
)
|
2013-05-19 01:04:03 +08:00
|
|
|
|
2013-07-23 17:15:20 +08:00
|
|
|
macro_rules! test_transpose_mat_impl(
|
|
|
|
($t: ty) => (
|
2013-11-27 18:16:16 +08:00
|
|
|
10000.times(|| {
|
2013-07-23 17:15:20 +08:00
|
|
|
let randmat : $t = random();
|
|
|
|
|
2013-10-14 16:22:32 +08:00
|
|
|
assert!(na::transpose(&na::transpose(&randmat)) == randmat);
|
2013-11-27 18:16:16 +08:00
|
|
|
})
|
2013-07-23 17:15:20 +08:00
|
|
|
);
|
|
|
|
)
|
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_transpose_mat1() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_transpose_mat_impl!(Mat1<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-07-23 17:15:20 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_transpose_mat2() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_transpose_mat_impl!(Mat2<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-07-23 17:15:20 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_transpose_mat3() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_transpose_mat_impl!(Mat3<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-07-23 17:15:20 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_transpose_mat4() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_transpose_mat_impl!(Mat4<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-07-23 17:15:20 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_transpose_mat5() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_transpose_mat_impl!(Mat5<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-07-23 17:15:20 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_transpose_mat6() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_transpose_mat_impl!(Mat6<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-07-23 17:15:20 +08:00
|
|
|
|
2013-05-19 01:04:03 +08:00
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_inv_mat1() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_inv_mat_impl!(Mat1<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-05-19 01:04:03 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_inv_mat2() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_inv_mat_impl!(Mat2<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-05-19 01:04:03 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_inv_mat3() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_inv_mat_impl!(Mat3<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-05-19 01:04:03 +08:00
|
|
|
|
2013-07-01 05:19:36 +08:00
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_inv_mat4() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_inv_mat_impl!(Mat4<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-07-01 05:19:36 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_inv_mat5() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_inv_mat_impl!(Mat5<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-07-01 05:19:36 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_inv_mat6() {
|
2013-10-18 04:40:44 +08:00
|
|
|
test_inv_mat_impl!(Mat6<f64>);
|
2013-08-05 16:13:44 +08:00
|
|
|
}
|
2013-05-19 01:04:03 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_rotation2() {
|
2013-11-27 18:16:16 +08:00
|
|
|
10000.times(|| {
|
2013-10-08 07:22:56 +08:00
|
|
|
let randmat: na::Rot2<f64> = na::one();
|
2013-10-14 17:22:38 +08:00
|
|
|
let ang = Vec1::new(abs::<f64>(random()) % Real::pi());
|
2013-05-19 19:53:19 +08:00
|
|
|
|
2014-01-10 03:48:30 +08:00
|
|
|
assert!(na::approx_eq(&na::rotation(&na::append_rotation(&randmat, &ang)), &ang));
|
2013-11-27 18:16:16 +08:00
|
|
|
})
|
2013-05-19 01:04:03 +08:00
|
|
|
}
|
2013-06-29 23:19:21 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_index_mat2() {
|
2013-10-18 04:40:44 +08:00
|
|
|
let mat: Mat2<f64> = random();
|
2013-06-29 23:19:21 +08:00
|
|
|
|
2013-10-14 16:22:32 +08:00
|
|
|
assert!(mat.at((0, 1)) == na::transpose(&mat).at((1, 0)));
|
2013-06-29 23:19:21 +08:00
|
|
|
}
|
2013-07-23 17:15:20 +08:00
|
|
|
|
|
|
|
#[test]
|
2013-08-05 16:13:44 +08:00
|
|
|
fn test_inv_rotation3() {
|
2013-11-27 18:16:16 +08:00
|
|
|
10000.times(|| {
|
2013-10-18 04:40:44 +08:00
|
|
|
let randmat: Rot3<f64> = na::one();
|
|
|
|
let dir: Vec3<f64> = random();
|
2013-10-14 16:22:32 +08:00
|
|
|
let ang = na::normalize(&dir) * (abs::<f64>(random()) % Real::pi());
|
|
|
|
let rot = na::append_rotation(&randmat, &ang);
|
2013-09-22 20:22:17 +08:00
|
|
|
|
2014-01-10 03:48:30 +08:00
|
|
|
assert!(na::approx_eq(&(na::transpose(&rot) * rot), &na::one()));
|
2013-11-27 18:16:16 +08:00
|
|
|
})
|
2013-09-22 20:22:17 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_mean_dmat() {
|
2013-10-18 04:40:44 +08:00
|
|
|
let mat = DMat::from_row_vec(
|
2013-09-22 20:22:17 +08:00
|
|
|
3,
|
|
|
|
3,
|
|
|
|
[
|
|
|
|
1.0f64, 2.0, 3.0,
|
|
|
|
4.0f64, 5.0, 6.0,
|
|
|
|
7.0f64, 8.0, 9.0,
|
|
|
|
]
|
|
|
|
);
|
|
|
|
|
2014-01-10 03:48:30 +08:00
|
|
|
assert!(na::approx_eq(&na::mean(&mat), &DVec::from_vec(3, [4.0f64, 5.0, 6.0])));
|
2013-09-22 20:22:17 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_cov_dmat() {
|
2013-10-18 04:40:44 +08:00
|
|
|
let mat = DMat::from_row_vec(
|
2013-09-22 20:22:17 +08:00
|
|
|
5,
|
|
|
|
3,
|
|
|
|
[
|
|
|
|
4.0, 2.0, 0.60,
|
|
|
|
4.2, 2.1, 0.59,
|
|
|
|
3.9, 2.0, 0.58,
|
|
|
|
4.3, 2.1, 0.62,
|
|
|
|
4.1, 2.2, 0.63
|
|
|
|
]
|
|
|
|
);
|
|
|
|
|
2013-10-18 04:40:44 +08:00
|
|
|
let expected = DMat::from_row_vec(
|
2013-09-22 20:22:17 +08:00
|
|
|
3,
|
|
|
|
3,
|
|
|
|
[
|
|
|
|
0.025, 0.0075, 0.00175,
|
|
|
|
0.0075, 0.007, 0.00135,
|
|
|
|
0.00175, 0.00135, 0.00043
|
|
|
|
]
|
|
|
|
);
|
|
|
|
|
2014-01-10 03:48:30 +08:00
|
|
|
assert!(na::approx_eq(&na::cov(&mat), &expected));
|
2013-07-23 17:15:20 +08:00
|
|
|
}
|
2013-10-18 04:40:44 +08:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_transpose_dmat() {
|
|
|
|
let mat = DMat::from_row_vec(
|
|
|
|
8,
|
|
|
|
4,
|
|
|
|
[
|
|
|
|
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
|
|
|
|
]
|
|
|
|
);
|
|
|
|
|
|
|
|
assert!(na::transpose(&na::transpose(&mat)) == mat);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_dmat_from_vec() {
|
|
|
|
let mat1 = DMat::from_row_vec(
|
|
|
|
8,
|
|
|
|
4,
|
|
|
|
[
|
|
|
|
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
|
|
|
|
]
|
|
|
|
);
|
|
|
|
|
|
|
|
let mat2 = DMat::from_col_vec(
|
|
|
|
8,
|
|
|
|
4,
|
|
|
|
[
|
|
|
|
1, 5, 9, 13, 17, 21, 25, 29,
|
|
|
|
2, 6, 10, 14, 18, 22, 26, 30,
|
|
|
|
3, 7, 11, 15, 19, 23, 27, 31,
|
|
|
|
4, 8, 12, 16, 20, 24, 28, 32
|
|
|
|
]
|
|
|
|
);
|
|
|
|
|
|
|
|
println!("mat1: {:?}, mat2: {:?}", mat1, mat2);
|
|
|
|
|
|
|
|
assert!(mat1 == mat2);
|
|
|
|
}
|