Skip to content

Commit 5c6e170

Browse files
simplified creation of complex double precision input data set. Relaxed relative tolerance a little bit
1 parent 8f75dec commit 5c6e170

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

mkl_fft/tests/test_fft1d.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -55,10 +55,7 @@ def setUp(self):
5555
rnd.seed(1234567)
5656
self.xd1 = rnd.standard_normal(128)
5757
self.xf1 = self.xd1.astype(np.float32)
58-
self.xz1 = np.dot(
59-
rnd.standard_normal((128,2)),
60-
np.array([1.0 + 0.0j, 0.0 + 1.0j], dtype=np.complex128)
61-
)
58+
self.xz1 = rnd.standard_normal((128,2)).view(dtype=np.complex128)
6259
self.xc1 = self.xz1.astype(np.complex64)
6360

6461
def test_vector1(self):
@@ -69,7 +66,7 @@ def test_vector1(self):
6966

7067
f1 = mkl_fft.fft(self.xc1)
7168
f2 = np_fft.fft(self.xc1)
72-
assert_allclose(f1,f2, rtol=1e-6, atol=2e-6)
69+
assert_allclose(f1,f2, rtol=2e-6, atol=2e-6)
7370

7471
def test_vector2(self):
7572
"ifft(fft(x)) is identity"

0 commit comments

Comments
 (0)