use custos::CPU;
use custos_math::{ClipOp, Matrix};

#[cfg(feature = "opencl")]
use custos::opencl::OpenCL;

#[cfg(feature = "cpu")]
#[test]
fn test_clip_cpu() {
    let device = CPU::new();

    let x = Matrix::<i32>::from((&device, (1, 5), [100, 10, 2000, -500, -5]));

    let res = device.clip(&x, -99, 99);
    assert_eq!(vec![99, 10, 99, -99, -5], res.read());
}

#[cfg(feature = "opencl")]
#[test]
fn test_clip_cl() {
    let device = OpenCL::new(0).unwrap();

    let x = Matrix::<i32, OpenCL>::from((&device, (1, 5), [100, 10, 2000, -500, -5]));

    let res = device.clip(&x, -99, 99);
    assert_eq!(vec![99, 10, 99, -99, -5], res.read());
}

#[cfg(feature = "cuda")]
#[test]
fn test_clip_cuda() {
    let device = custos::CUDA::new(0).unwrap();

    let x = Matrix::from((&device, (1, 5), [100, 10, 2000, -500, -5]));

    let res = device.clip(&x, -99, 99);
    assert_eq!(vec![99, 10, 99, -99, -5], res.read());
}