You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.6 KiB
Rust
58 lines
1.6 KiB
Rust
use num_traits::Float;
|
|
use std::iter::zip;
|
|
use vector_victor::Matrix;
|
|
|
|
pub trait Approx: PartialEq {
|
|
fn approx(left: &Self, right: &Self) -> bool {
|
|
left == right
|
|
}
|
|
}
|
|
|
|
macro_rules! multi_impl { ($name:ident for $($t:ty),*) => ($( impl $name for $t {} )*) }
|
|
multi_impl!(Approx for i8, i16, i32, i64, i128, isize, u8, u16, u32, u64, u128, usize);
|
|
|
|
impl Approx for f32 {
|
|
fn approx(left: &f32, right: &f32) -> bool {
|
|
f32::abs(left - right) <= f32::epsilon()
|
|
}
|
|
}
|
|
|
|
impl Approx for f64 {
|
|
fn approx(left: &f64, right: &f64) -> bool {
|
|
f64::abs(left - right) <= f32::epsilon() as f64
|
|
}
|
|
}
|
|
|
|
impl<T: Copy + Approx, const M: usize, const N: usize> Approx for Matrix<T, M, N> {
|
|
fn approx(left: &Self, right: &Self) -> bool {
|
|
zip(left.elements(), right.elements()).all(|(l, r)| T::approx(l, r))
|
|
}
|
|
}
|
|
|
|
pub fn approx<T: Approx>(left: &T, right: &T) -> bool {
|
|
T::approx(left, right)
|
|
}
|
|
|
|
macro_rules! assert_approx {
|
|
($left:expr, $right:expr $(,)?) => {
|
|
match (&$left, &$right) {
|
|
(_left_val, _right_val) => {
|
|
assert_approx!($left, $right, "Difference is less than epsilon")
|
|
}
|
|
}
|
|
};
|
|
($left:expr, $right:expr, $($arg:tt)+) => {
|
|
match (&$left, &$right) {
|
|
(left_val, right_val) => {
|
|
pub fn approx<T: Approx>(left: &T, right: &T) -> bool {
|
|
T::approx(left, right)
|
|
}
|
|
|
|
if !approx(left_val, right_val){
|
|
assert_eq!(left_val, right_val, $($arg)+) // done this way to get nice errors
|
|
}
|
|
}
|
|
}
|
|
};
|
|
}
|