Example #1
0
 def test_fft_pair(self):
     my_rfft, my_irfft = rfft(self.x, nfft=32, fft_pair=True)
     x_fft = my_rfft(self.x)
     assert_array_almost_equal(my_irfft(x_fft), pad_array(self.x, (32,)))
Example #2
0
 def test_fft_2d_nfft(self):
     my_rfft = rfft(self.x2, nfft=32)
     assert_array_almost_equal(my_rfft(self.x2), np.fft.rfft(self.x2, n=32))
Example #3
0
 def test_fft_2d_default(self):
     my_rfft = rfft(self.x2)
     assert_array_almost_equal(my_rfft(self.x2), np.fft.rfft(self.x2))
Example #4
0
 def test_fft_2d_axis(self):
     my_rfft = rfft(self.x2, axis=0)
     assert_array_almost_equal(my_rfft(self.x2),
                               np.fft.rfft(self.x2, axis=0))
Example #5
0
 def test_fft_1d(self):
     my_rfft = rfft(self.x)
     assert_array_almost_equal(my_rfft(self.x), np.fft.rfft(self.x))