Example #1
0
 def test_fft_pair_cropped(self):
     my_fft, my_ifft = fft(self.x, nfft=32, fft_pair=True, crop_ifft=True)
     x_fft = my_fft(self.x)
     assert_array_almost_equal(my_ifft(x_fft), self.x)
Example #2
0
 def test_fft_2d_nfft(self):     # No additional coverage for this, keep?
     my_fft = fft(self.x2, nfft=32)
     assert_array_almost_equal(my_fft(self.x2), np.fft.fft(self.x2, n=32))
Example #3
0
 def test_fft_pair(self):
     my_fft, my_ifft = fft(self.x, nfft=32, fft_pair=True)
     x_fft = my_fft(self.x)
     assert_array_almost_equal(my_ifft(x_fft), pad_array(self.x, (32,)))
Example #4
0
 def test_fft_2d_default(self):
     my_fft = fft(self.x2)
     assert_array_almost_equal(my_fft(self.x2), np.fft.fft(self.x2))
Example #5
0
 def test_fft_2d_axis(self):
     my_fft = fft(self.x2, axis=0)
     assert_array_almost_equal(my_fft(self.x2), np.fft.fft(self.x2, axis=0))
Example #6
0
 def test_fft_nfft(self):
     my_fft = fft(self.x, nfft=32)
     assert_array_almost_equal(my_fft(self.x), np.fft.fft(self.x, n=32))
Example #7
0
 def test_fft_nfft_small(self):
     my_fft = fft(self.x, nfft=7)
     assert_array_almost_equal(my_fft(self.x), np.fft.fft(self.x, n=7))
Example #8
0
 def test_fft_2d_nfft(self):  # No additional coverage for this, keep?
     my_fft = fft(self.x2, nfft=32)
     assert_array_almost_equal(my_fft(self.x2), np.fft.fft(self.x2, n=32))
Example #9
0
 def test_fft_1d(self):
     my_fft = fft(self.x)
     assert_array_almost_equal(my_fft(self.x), np.fft.fft(self.x))
Example #10
0
 def test_fft_2d_axis(self):
     my_fft = fft(self.x2, axis=0)
     assert_array_almost_equal(my_fft(self.x2), np.fft.fft(self.x2, axis=0))
Example #11
0
 def test_fft_2d_default(self):
     my_fft = fft(self.x2)
     assert_array_almost_equal(my_fft(self.x2), np.fft.fft(self.x2))
Example #12
0
 def test_fft_nfft_small(self):
     my_fft = fft(self.x, nfft=7)
     assert_array_almost_equal(my_fft(self.x), np.fft.fft(self.x, n=7))
Example #13
0
 def test_fft_nfft(self):
     my_fft = fft(self.x, nfft=32)
     assert_array_almost_equal(my_fft(self.x), np.fft.fft(self.x, n=32))
Example #14
0
 def test_fft_1d(self):
     my_fft = fft(self.x)
     assert_array_almost_equal(my_fft(self.x), np.fft.fft(self.x))