Skip to content

Commit 5262d8d

Browse files
fixed the broken test
1 parent 5c6e170 commit 5262d8d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mkl_fft/tests/test_fft1d.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +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 = rnd.standard_normal((128,2)).view(dtype=np.complex128)
58+
self.xz1 = rnd.standard_normal((128,2)).view(dtype=np.complex128).squeeze()
5959
self.xc1 = self.xz1.astype(np.complex64)
6060

6161
def test_vector1(self):
@@ -160,7 +160,7 @@ def test_vector9(self):
160160
assert_allclose(f1, f2)
161161

162162
def test_vector10(self):
163-
"check n"
163+
"check n for real arrays"
164164
x = self.xd1[:8].copy()
165165
f1 = mkl_fft.fft(x, n = 7)
166166
f2 = mkl_fft.fft(self.xd1[:7])
@@ -173,7 +173,7 @@ def test_vector10(self):
173173
assert_allclose(f1, f2)
174174

175175
def test_vector11(self):
176-
"check n"
176+
"check n for complex arrays"
177177
x = self.xz1[:8].copy()
178178
f1 = mkl_fft.fft(x, n = 7)
179179
f2 = mkl_fft.fft(self.xz1[:7])

0 commit comments

Comments
 (0)