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))
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))
def test_fft_1d(self):
my_rfft = rfft(self.x)
assert_array_almost_equal(my_rfft(self.x), np.fft.rfft(self.x))
def test_fft_2d_default(self):
my_rfft = rfft(self.x2)
assert_array_almost_equal(my_rfft(self.x2), np.fft.rfft(self.x2))
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,)))
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))
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))
def test_fft_2d_default(self):
my_rfft = rfft(self.x2)
assert_array_almost_equal(my_rfft(self.x2), np.fft.rfft(self.x2))
def test_fft_1d(self):
my_rfft = rfft(self.x)
assert_array_almost_equal(my_rfft(self.x), np.fft.rfft(self.x))