1#include <gtest/gtest.h>
 2
 3#include "../spqlios/arithmetic/vec_znx_arithmetic_private.h"
 4#include "testlib/fft64_dft.h"
 5#include "testlib/fft64_layouts.h"
 6#include "testlib/polynomial_vector.h"
 7
 8// todo: remove when registered
 9typedef typeof(fft64_svp_prepare_ref) SVP_PREPARE_F;
10
11void test_fft64_svp_prepare(SVP_PREPARE_F svp_prepare) {
12  for (uint64_t n : {2, 4, 8, 64, 128}) {
13    MODULE* module = new_module_info(n, FFT64);
14    znx_i64 in = znx_i64::random_log2bound(n, 40);
15    fft64_svp_ppol_layout out(n);
16    reim_fft64vec expect = simple_fft64(in);
17    svp_prepare(module, out.data, in.data());
18    const double* ed = (double*)expect.data();
19    const double* ac = (double*)out.data;
20    for (uint64_t i = 0; i < n; ++i) {
21      ASSERT_LE(abs(ed[i] - ac[i]), 1e-10) << i << n;
22    }
23    delete_module_info(module);
24  }
25}
26
27TEST(svp_prepare, fft64_svp_prepare_ref) { test_fft64_svp_prepare(fft64_svp_prepare_ref); }
28TEST(svp_prepare, svp_prepare) { test_fft64_svp_prepare(svp_prepare); }