diff --git a/src/tests/algorithms/FFT/test-fft-data.h b/src/tests/algorithms/FFT/test-fft-data.h index b94fe266c6f87b117f84dce294afe1b826a4d8c9..bde92390acf1f805a4ca34d385dcdad181c710ea 100644 --- a/src/tests/algorithms/FFT/test-fft-data.h +++ b/src/tests/algorithms/FFT/test-fft-data.h @@ -48,18 +48,17 @@ static const float SAMPLE_RATE = 64.0; // ------------------------ EXPECTED OUTPUT ------------------------ static const std::vector<float> INTENSITY{ - 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, - 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, - 1.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, 0.000000, - 0.000000, 0.000000, 0.000000, 0.000000, 2.000000, 0.000000, 0.000000, - 0.000000, 0.000000, 0.000000, 0.000000}; + 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.5, 0., + 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., + 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., + 0., 0., 0.5, 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.}; static const std::vector<float> FREQUENCY{ - 0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, - 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, - 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, -32.0, -31.0, -30.0, -29.0, - -28.0, -27.0, -26.0, -25.0, -24.0, -23.0, -22.0, -21.0, -20.0, -19.0, -18.0, - -17.0, -16.0, -15.0, -14.0, -13.0, -12.0, -11.0, -10.0, -9.0, -8.0, -7.0, - -6.0, -5.0, -4.0, -3.0, -2.0, -1.0}; + 0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, + 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, + 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, -32.0, + -31.0, -30.0, -29.0, -28.0, -27.0, -26.0, -25.0, -24.0, -23.0, -22.0, -21.0, + -20.0, -19.0, -18.0, -17.0, -16.0, -15.0, -14.0, -13.0, -12.0, -11.0, -10.0, + -9.0, -8.0, -7.0, -6.0, -5.0, -4.0, -3.0, -2.0, -1.0}; } // namespace Boardcore diff --git a/src/tests/algorithms/FFT/test-fft.cpp b/src/tests/algorithms/FFT/test-fft.cpp index 3695878608c2245a373af619a8502acff8548189..01773e8e2a38fe51526b1387105fa19baf50ed89 100644 --- a/src/tests/algorithms/FFT/test-fft.cpp +++ b/src/tests/algorithms/FFT/test-fft.cpp @@ -46,9 +46,9 @@ int main() FFT<SAMPLES>::fft(signal_vector); Vector<float, SAMPLES> fft_freq = FFT<SAMPLES>::fftfreq(SAMPLE_RATE); - for (int i = 0; i < fft_result.size() / 2; i++) + for (int i = 0; i < fft_result.size(); i++) { - if (std::abs(2.0 / SAMPLES * std::abs(fft_result(i)) - INTENSITY[i]) > + if (std::abs(1.0 / SAMPLES * std::abs(fft_result(i)) - INTENSITY[i]) > 0.001) { printf("FFT result differs from the correct one [%d]: %f != %f\n",