Check if fft_filter test passes with complex128

ptycho · Dec 13, 2024 · c20dfb6 · c20dfb6
1 parent 40dbf12
commit c20dfb6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)