use rustfatigue::{abs_difference, eq_load, get_amplitudes, get_peaktrough};

static INP: [f64; 20] = [
    40.0, 15.0, 72.0, 22.0, 43.0, 82.0, 75.0, 7.0, 34.0, 49.0, 95.0, 75.0, 85.0, 47.0, 63.0, 31.0,
    90.0, 20.0, 37.0, 39.0,
];

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn peaktrough_no_threshold() {
        let ans: [f64; 15] = [
            40.0, 15.0, 72.0, 22.0, 82.0, 7.0, 95.0, 75.0, 85.0, 47.0, 63.0, 31.0, 90.0, 20.0, 39.0,
        ];
        assert!(get_peaktrough(&INP.to_vec(), 0.0)
            .iter()
            .enumerate()
            .all(|(i, &val)| val == ans[i]));
    }
    #[test]
    fn peaktrough_with_threshold() {
        let ans: [f64; 10] = [40.0, 15.0, 72.0, 22.0, 82.0, 7.0, 95.0, 31.0, 90.0, 20.0];

        assert!(get_peaktrough(&INP.to_vec(), 20.0)
            .iter()
            .enumerate()
            .all(|(i, &val)| val == ans[i]));
    }
    #[test]
    fn amptest() {
        let ans: [f64; 14] = [
            50.0, 50.0, 10.0, 10.0, 16.0, 16.0, 59.0, 59.0, 25.0, 67.0, 75.0, 88.0, 75.0, 19.0,
        ];
        let peaktroughs = get_peaktrough(&INP.to_vec(), 0.0);
        assert!(get_amplitudes(&peaktroughs)
            .iter()
            .enumerate()
            .all(|(i, &val)| val == ans[i]));
    }

    #[test]
    fn eq_load_float() {
        let out = eq_load(&INP.to_vec(), 4.0, 20);
        println!("{}", out);
        assert!(abs_difference(out, 54.83366824817066) < 0.0001);
    }
}