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_rfftn_pair_cropped(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,
crop_ifft=True)
x_fft = my_rfftn(self.x3)
assert_array_almost_equal(my_irfftn(x_fft), self.x3)
def test_rfftn_3d_axes(self):
test_axes = (0, -1)
my_rfftn = rfftn(self.x3, axes=test_axes)
assert_array_almost_equal(my_rfftn(self.x3),
np.fft.rfftn(self.x3, axes=test_axes))
def test_rfftn_3d_mixed_and_short_shape(self):
test_shape = (2, 16)
my_rfftn = rfftn(self.x3, shape=test_shape)
assert_array_almost_equal(my_rfftn(self.x3),
np.fft.rfftn(self.x3, s=test_shape))
def test_rfftn_3d_smaller_shape(self):
test_shape = (3, 3, 6)
my_rfftn = rfftn(self.x3, shape=test_shape)
assert_array_almost_equal(my_rfftn(self.x3),
np.fft.rfftn(self.x3, s=test_shape))
def test_rfftn_3d_larger_shape(self):
test_shape = (4, 6, 14)
my_rfftn = rfftn(self.x3, shape=test_shape)
assert_array_almost_equal(my_rfftn(self.x3),
np.fft.rfftn(self.x3, s=test_shape))
def test_rfftn_2d(self):
my_rfftn = rfftn(self.x2)
assert_array_almost_equal(my_rfftn(self.x2), np.fft.rfftn(self.x2))
def test_rfftn_3d_axes(self):
test_axes = (0, -1)
my_rfftn = rfftn(self.x3, axes=test_axes)
assert_array_almost_equal(my_rfftn(self.x3),
np.fft.rfftn(self.x3, axes=test_axes))
def test_rfftn_3d_mixed_and_short_shape(self):
test_shape = (2, 16)
my_rfftn = rfftn(self.x3, shape=test_shape)
assert_array_almost_equal(my_rfftn(self.x3), np.fft.rfftn(self.x3,
s=test_shape))
def test_rfftn_3d_smaller_shape(self):
test_shape = (3, 3, 6)
my_rfftn = rfftn(self.x3, shape=test_shape)
assert_array_almost_equal(my_rfftn(self.x3), np.fft.rfftn(self.x3,
s=test_shape))
def test_rfftn_3d_larger_shape(self):
test_shape = (4, 6, 14)
my_rfftn = rfftn(self.x3, shape=test_shape)
assert_array_almost_equal(my_rfftn(self.x3), np.fft.rfftn(self.x3,
s=test_shape))
def test_rfftn_2d(self):
my_rfftn = rfftn(self.x2)
assert_array_almost_equal(my_rfftn(self.x2), np.fft.rfftn(self.x2))