From c20dfb60da9ebb8f53af7c754756816e3e52dca0 Mon Sep 17 00:00:00 2001 From: Benedikt Daurer Date: Fri, 13 Dec 2024 19:00:10 +0000 Subject: [PATCH] Check if fft_filter test passes with complex128 --- 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 364352a5c..0d6e3d719 100644 --- a/test/accelerate_tests/base_tests/array_utils_test.py +++ b/test/accelerate_tests/base_tests/array_utils_test.py @@ -292,7 +292,7 @@ def test_crop_pad_3(self): np.testing.assert_array_almost_equal(A, exp_A) def test_fft_filter(self): - data = np.zeros((256, 512), dtype=COMPLEX_TYPE) + data = np.zeros((256, 512), dtype=np.complex128) data[64:-64,128:-128] = 1 + 1.j prefactor = np.zeros_like(data) @@ -343,7 +343,7 @@ def test_fft_filter(self): np.testing.assert_array_almost_equal(output.flat[::2000], known_test_output) def test_fft_filter_batched(self): - data = np.zeros((2,256, 512), dtype=COMPLEX_TYPE) + data = np.zeros((2,256, 512), dtype=np.complex128) data[:,64:-64,128:-128] = 1 + 1.j prefactor = np.zeros_like(data)