File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -55,7 +55,7 @@ def setUp(self):
55
55
rnd .seed (1234567 )
56
56
self .xd1 = rnd .standard_normal (128 )
57
57
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 ()
59
59
self .xc1 = self .xz1 .astype (np .complex64 )
60
60
61
61
def test_vector1 (self ):
@@ -160,7 +160,7 @@ def test_vector9(self):
160
160
assert_allclose (f1 , f2 )
161
161
162
162
def test_vector10 (self ):
163
- "check n"
163
+ "check n for real arrays "
164
164
x = self .xd1 [:8 ].copy ()
165
165
f1 = mkl_fft .fft (x , n = 7 )
166
166
f2 = mkl_fft .fft (self .xd1 [:7 ])
@@ -173,7 +173,7 @@ def test_vector10(self):
173
173
assert_allclose (f1 , f2 )
174
174
175
175
def test_vector11 (self ):
176
- "check n"
176
+ "check n for complex arrays "
177
177
x = self .xz1 [:8 ].copy ()
178
178
f1 = mkl_fft .fft (x , n = 7 )
179
179
f2 = mkl_fft .fft (self .xz1 [:7 ])
You can’t perform that action at this time.
0 commit comments