forked from M-Labs/nalgebra
368 lines
8.1 KiB
Rust
368 lines
8.1 KiB
Rust
extern crate rand;
|
|
extern crate typenum;
|
|
extern crate nalgebra as na;
|
|
|
|
use rand::random;
|
|
use typenum::U10;
|
|
use na::{VecN, Vec1, Vec2, Vec3, Vec4, Vec5, Vec6, Mat3, Rot2, Rot3, Iterable, IterableMut};
|
|
|
|
macro_rules! test_iterator_impl(
|
|
($t: ty, $n: ty) => (
|
|
for _ in 0usize .. 10000 {
|
|
let v: $t = random();
|
|
let mut mv: $t = v.clone();
|
|
let n: $n = random();
|
|
|
|
let nv: $t = v.iter().map(|e| *e * n).collect();
|
|
|
|
for e in mv.iter_mut() {
|
|
*e = *e * n
|
|
}
|
|
|
|
assert!(nv == mv && nv == v * n);
|
|
}
|
|
)
|
|
);
|
|
|
|
macro_rules! test_commut_dot_impl(
|
|
($t: ty) => (
|
|
for _ in 0usize .. 10000 {
|
|
let v1 : $t = random();
|
|
let v2 : $t = random();
|
|
|
|
assert!(na::approx_eq(&na::dot(&v1, &v2), &na::dot(&v2, &v1)));
|
|
}
|
|
);
|
|
);
|
|
|
|
macro_rules! test_scalar_op_impl(
|
|
($t: ty, $n: ty) => (
|
|
for _ in 0usize .. 10000 {
|
|
let v1 : $t = random();
|
|
let n : $n = random();
|
|
|
|
assert!(na::approx_eq(&((v1 * n) / n), &v1));
|
|
assert!(na::approx_eq(&((v1 / n) * n), &v1));
|
|
assert!(na::approx_eq(&((v1 - n) + n), &v1));
|
|
assert!(na::approx_eq(&((v1 + n) - n), &v1));
|
|
|
|
let mut v1 : $t = random();
|
|
let v0 : $t = v1.clone();
|
|
let n : $n = random();
|
|
|
|
v1 = v1 * n;
|
|
v1 = v1 / n;
|
|
|
|
assert!(na::approx_eq(&v1, &v0));
|
|
}
|
|
);
|
|
);
|
|
|
|
macro_rules! test_basis_impl(
|
|
($t: ty) => (
|
|
for _ in 0usize .. 10000 {
|
|
na::canonical_basis(|e1: $t| {
|
|
na::canonical_basis(|e2: $t| {
|
|
assert!(e1 == e2 || na::approx_eq(&na::dot(&e1, &e2), &na::zero()));
|
|
|
|
true
|
|
});
|
|
|
|
assert!(na::approx_eq(&na::norm(&e1), &na::one()));
|
|
|
|
true
|
|
})
|
|
}
|
|
);
|
|
);
|
|
|
|
macro_rules! test_subspace_basis_impl(
|
|
($t: ty) => (
|
|
for _ in 0usize .. 10000 {
|
|
let v : $t = random();
|
|
let v1 = na::normalize(&v);
|
|
|
|
na::orthonormal_subspace_basis(&v1, |e1| {
|
|
// check vectors are orthogonal to v1
|
|
assert!(na::approx_eq(&na::dot(&v1, &e1), &na::zero()));
|
|
// check vectors form an orthonormal basis
|
|
assert!(na::approx_eq(&na::norm(&e1), &na::one()));
|
|
// check vectors form an ortogonal basis
|
|
na::orthonormal_subspace_basis(&v1, |e2| {
|
|
assert!(e1 == e2 || na::approx_eq(&na::dot(&e1, &e2), &na::zero()));
|
|
|
|
true
|
|
});
|
|
|
|
true
|
|
})
|
|
}
|
|
);
|
|
);
|
|
|
|
#[test]
|
|
fn test_cross_vec3() {
|
|
for _ in 0usize .. 10000 {
|
|
let v1 : Vec3<f64> = random();
|
|
let v2 : Vec3<f64> = random();
|
|
let v3 : Vec3<f64> = na::cross(&v1, &v2);
|
|
|
|
assert!(na::approx_eq(&na::dot(&v3, &v2), &na::zero()));
|
|
assert!(na::approx_eq(&na::dot(&v3, &v1), &na::zero()));
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_commut_dot_vec1() {
|
|
test_commut_dot_impl!(Vec1<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_commut_dot_vec2() {
|
|
test_commut_dot_impl!(Vec2<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_commut_dot_vec3() {
|
|
test_commut_dot_impl!(Vec3<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_commut_dot_vec4() {
|
|
test_commut_dot_impl!(Vec4<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_commut_dot_vec5() {
|
|
test_commut_dot_impl!(Vec5<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_commut_dot_vec6() {
|
|
test_commut_dot_impl!(Vec6<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_basis_vec1() {
|
|
test_basis_impl!(Vec1<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_basis_vec2() {
|
|
test_basis_impl!(Vec2<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_basis_vec3() {
|
|
test_basis_impl!(Vec3<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_basis_vec4() {
|
|
test_basis_impl!(Vec4<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_basis_vec5() {
|
|
test_basis_impl!(Vec5<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_basis_vec6() {
|
|
test_basis_impl!(Vec6<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_subspace_basis_vec1() {
|
|
test_subspace_basis_impl!(Vec1<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_subspace_basis_vec2() {
|
|
test_subspace_basis_impl!(Vec2<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_subspace_basis_vec3() {
|
|
test_subspace_basis_impl!(Vec3<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_subspace_basis_vec4() {
|
|
test_subspace_basis_impl!(Vec4<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_subspace_basis_vec5() {
|
|
test_subspace_basis_impl!(Vec5<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_subspace_basis_vec6() {
|
|
test_subspace_basis_impl!(Vec6<f64>);
|
|
}
|
|
|
|
#[test]
|
|
fn test_scalar_op_vec1() {
|
|
test_scalar_op_impl!(Vec1<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_scalar_op_vec2() {
|
|
test_scalar_op_impl!(Vec2<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_scalar_op_vec3() {
|
|
test_scalar_op_impl!(Vec3<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_scalar_op_vec4() {
|
|
test_scalar_op_impl!(Vec4<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_scalar_op_vec5() {
|
|
test_scalar_op_impl!(Vec5<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_scalar_op_vec6() {
|
|
test_scalar_op_impl!(Vec6<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_iterator_vec1() {
|
|
test_iterator_impl!(Vec1<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_iterator_vec2() {
|
|
test_iterator_impl!(Vec2<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_iterator_vec3() {
|
|
test_iterator_impl!(Vec3<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_iterator_vec4() {
|
|
test_iterator_impl!(Vec4<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_iterator_vec5() {
|
|
test_iterator_impl!(Vec5<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_iterator_vec6() {
|
|
test_iterator_impl!(Vec6<f64>, f64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_ord_vec3() {
|
|
// equality
|
|
assert!(Vec3::new(0.5f64, 0.5, 0.5) == Vec3::new(0.5, 0.5, 0.5));
|
|
assert!(!(Vec3::new(1.5f64, 0.5, 0.5) == Vec3::new(0.5, 0.5, 0.5)));
|
|
assert!(Vec3::new(1.5f64, 0.5, 0.5) != Vec3::new(0.5, 0.5, 0.5));
|
|
|
|
// comparable
|
|
assert!(na::partial_cmp(&Vec3::new(0.5f64, 0.3, 0.3), &Vec3::new(1.0, 2.0, 1.0)).is_le());
|
|
assert!(na::partial_cmp(&Vec3::new(0.5f64, 0.3, 0.3), &Vec3::new(1.0, 2.0, 1.0)).is_lt());
|
|
assert!(na::partial_cmp(&Vec3::new(2.0f64, 4.0, 2.0), &Vec3::new(1.0, 2.0, 1.0)).is_ge());
|
|
assert!(na::partial_cmp(&Vec3::new(2.0f64, 4.0, 2.0), &Vec3::new(1.0, 2.0, 1.0)).is_gt());
|
|
|
|
// not comparable
|
|
assert!(na::partial_cmp(&Vec3::new(0.0f64, 3.0, 0.0), &Vec3::new(1.0, 2.0, 1.0)).is_not_comparable());
|
|
}
|
|
|
|
#[test]
|
|
fn test_min_max_vec3() {
|
|
assert_eq!(na::sup(&Vec3::new(1.0f64, 2.0, 3.0), &Vec3::new(3.0, 2.0, 1.0)), Vec3::new(3.0, 2.0, 3.0));
|
|
assert_eq!(na::inf(&Vec3::new(1.0f64, 2.0, 3.0), &Vec3::new(3.0, 2.0, 1.0)), Vec3::new(1.0, 2.0, 1.0));
|
|
}
|
|
|
|
#[test]
|
|
fn test_outer_vec3() {
|
|
assert_eq!(
|
|
na::outer(&Vec3::new(1.0f64, 2.0, 3.0), &Vec3::new(4.0, 5.0, 6.0)),
|
|
Mat3::new(
|
|
4.0, 5.0, 6.0,
|
|
8.0, 10.0, 12.0,
|
|
12.0, 15.0, 18.0));
|
|
}
|
|
|
|
#[test]
|
|
fn test_vecn10_add_mul() {
|
|
for _ in 0usize .. 10000 {
|
|
let v1: VecN<f64, U10> = random();
|
|
|
|
assert!(na::approx_eq(&(v1 + v1), &(v1 * 2.0)))
|
|
}
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn test_vec3_rotation_between() {
|
|
for _ in 0usize .. 10000 {
|
|
let v1: Vec3<f64> = random();
|
|
|
|
let mut v2: Vec3<f64> = random();
|
|
v2 = na::normalize(&v2) * na::norm(&v1);
|
|
|
|
let rot = na::rotation_between(&v1, &v2);
|
|
|
|
assert!(na::approx_eq(&(rot * v1), &v2))
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_vec3_angle_between() {
|
|
for _ in 0usize .. 10000 {
|
|
let vec: Vec3<f64> = random();
|
|
let other: Vec3<f64> = random();
|
|
|
|
// Ensure the axis we are using is orthogonal to `vec`.
|
|
let axis_ang = na::cross(&vec, &other);
|
|
let ang = na::norm(&axis_ang);
|
|
let rot = Rot3::new(axis_ang);
|
|
|
|
let delta = na::angle_between(&vec, &(rot * vec));
|
|
|
|
assert!(na::approx_eq(&ang, &delta))
|
|
}
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn test_vec2_rotation_between() {
|
|
for _ in 0usize .. 10000 {
|
|
let v1: Vec2<f64> = random();
|
|
|
|
let mut v2: Vec2<f64> = random();
|
|
v2 = na::normalize(&v2) * na::norm(&v1);
|
|
|
|
let rot = na::rotation_between(&v1, &v2);
|
|
|
|
assert!(na::approx_eq(&(rot * v1), &v2))
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_vec2_angle_between() {
|
|
for _ in 0usize .. 10000 {
|
|
let axis_ang: Vec1<f64> = random();
|
|
let ang = na::norm(&axis_ang);
|
|
|
|
let rot: Rot2<f64> = Rot2::new(axis_ang);
|
|
let vec: Vec2<f64> = random();
|
|
|
|
let delta = na::angle_between(&vec, &(rot * vec));
|
|
|
|
assert!(na::approx_eq(&ang, &delta))
|
|
}
|
|
}
|