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

use std::fmt::Debug;

use tea_dtype::{IsNone, Number};

use crate::prelude::{Vec1View, EPS};

pub fn assert_vec1d_equal_numeric<T: IsNone + Debug, V1: Vec1View<T>, V2: Vec1View<T>>(
    v1: &V1,
    v2: &V2,
    epsilon: Option<f64>,
) where
    T::Inner: Number,
{
    assert_eq!(v1.len(), v2.len());
    let epsilon = epsilon.unwrap_or(EPS);
    for (x, y) in v1.titer().zip(v2.titer()) {
        if x.is_none() && y.is_none() {
            continue;
        } else if x.is_none() || y.is_none() {
            panic!(
                "Vectors are not approximately equal, x: {:?}, y: {:?}",
                x, y
            );
        } else {
            let x = x.unwrap().f64();
            let y = y.unwrap().f64();
            if !(x.is_nan() && y.is_nan()) {
                assert!(
                    (x - y).abs() < epsilon,
                    "Vectors are not approximately equal, x: {}, y: {}",
                    x,
                    y
                );
            } else if x.is_nan() && y.is_nan() {
                continue;
            } else {
                panic!(
                    "Vectors are not approximately equal, x: {:?}, y: {:?}",
                    x, y
                );
            }
        }
    }
}