From 939293edbd5bff8d22498adb3ee37e56f33c74e2 Mon Sep 17 00:00:00 2001 From: Benedikt Daurer Date: Fri, 13 Dec 2024 19:32:08 +0000 Subject: [PATCH] Adjust expected precision in fft filter tests --- test/accelerate_tests/base_tests/array_utils_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/accelerate_tests/base_tests/array_utils_test.py b/test/accelerate_tests/base_tests/array_utils_test.py index 7d59c43be..518ddbb5c 100644 --- a/test/accelerate_tests/base_tests/array_utils_test.py +++ b/test/accelerate_tests/base_tests/array_utils_test.py @@ -340,7 +340,7 @@ def test_fft_filter(self): -9.243618e-07-3.862528e-05j, -8.814586e-06-9.899617e-06j, -1.578241e-06+1.015331e-05j, -4.759308e-06+2.963323e-05j]) - np.testing.assert_array_almost_equal(output.flat[::2000], known_test_output) + np.testing.assert_array_almost_equal(output.flat[::2000], known_test_output, decimal=5) def test_fft_filter_batched(self): data = np.zeros((2,256, 512), dtype=COMPLEX_TYPE) @@ -391,7 +391,7 @@ def test_fft_filter_batched(self): -9.243618e-07-3.862528e-05j, -8.814586e-06-9.899617e-06j, -1.578241e-06+1.015331e-05j, -4.759308e-06+2.963323e-05j]) - np.testing.assert_array_almost_equal(output[1].flat[::2000], known_test_output) + np.testing.assert_array_almost_equal(output[1].flat[::2000], known_test_output, decimal=5) def test_complex_gaussian_filter_fft(self):