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):
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 ])
You can’t perform that action at this time.
0 commit comments