def test_rfftn_pair(self):
test_axes = (0, -1)
test_shape = (4, 16)
my_rfftn, my_irfftn = rfftn_pair(self.x3,
shape=test_shape,
axes=test_axes)
x_fft = my_rfftn(self.x3)
assert_array_almost_equal(my_irfftn(x_fft),
pad_array(self.x3, (4, 3, 16)))
def test_rfftn_pair(self):
test_axes = (0, -1)
test_shape = (4, 16)
my_rfftn, my_irfftn = rfftn(self.x3,
shape=test_shape,
axes=test_axes,
fft_pair=True)
x_fft = my_rfftn(self.x3)
assert_array_almost_equal(my_irfftn(x_fft),
pad_array(self.x3, (4, 3, 16)))
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_pair(self):
my_rfft, my_irfft = rfft_pair(self.x, nfft=32)
x_fft = my_rfft(self.x)
assert_array_almost_equal(my_irfft(x_fft), pad_array(self.x, (32, )))