use poulpy_cpu_ref::FFT64Ref;
use poulpy_hal::{api::ModuleNew, layouts::Module};

use crate::tfhe::blind_rotation::{
    CGGI,
    tests::test_suite::{
        generic_blind_rotation::test_blind_rotation,
        generic_lut::{test_lut_extended, test_lut_standard},
    },
};

#[test]
fn lut_standard() {
    let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(32);
    test_lut_standard(&module);
}

#[test]
fn lut_extended() {
    let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(32);
    test_lut_extended(&module);
}

#[test]
fn standard() {
    let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(512);
    test_blind_rotation::<CGGI, _, FFT64Ref>(&module, 224, 1, 1);
}

#[test]
fn block_binary() {
    let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(512);
    test_blind_rotation::<CGGI, _, FFT64Ref>(&module, 224, 7, 1);
}

#[test]
fn block_binary_extended() {
    let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(512);
    test_blind_rotation::<CGGI, _, FFT64Ref>(&module, 224, 7, 2);
}