def test_ifftn_pair_cropped(self):
my_ifftn, my_fftn = ifftn(self.x3_fft,
shape=(8, 4, 32),
fft_pair=True,
crop_fft=True)
my_x = my_ifftn(self.x3_fft)
assert_array_almost_equal(my_fftn(my_x),
self.x3_fft)
def test_ifftn_3d_axes(self):
test_axes = (0, -1)
my_ifftn = ifftn(self.x3_fft, axes=test_axes)
assert_array_almost_equal(my_ifftn(self.x3_fft),
np.fft.ifftn(self.x3_fft, axes=test_axes))
def test_ifftn_3d_mixed_and_short_shape(self):
test_shape = (2, 16)
my_ifftn = ifftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_ifftn(self.x3_fft),
np.fft.ifftn(self.x3_fft, s=test_shape))
def test_ifftn_3d_smaller_shape(self):
test_shape = (3, 3, 6)
my_ifftn = ifftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_ifftn(self.x3_fft),
np.fft.ifftn(self.x3_fft, s=test_shape))
def test_ifftn_3d_larger_shape(self):
test_shape = (4, 6, 14)
my_ifftn = ifftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_ifftn(self.x3_fft),
np.fft.ifftn(self.x3_fft, s=test_shape))
def test_ifftn_2d(self):
my_ifftn = ifftn(self.x2_fft)
assert_array_almost_equal(my_ifftn(self.x2_fft),
np.fft.ifftn(self.x2_fft))
def test_ifftn_pair(self):
my_ifftn, my_fftn = ifftn(self.x3_fft,
fft_pair=True)
my_x = my_ifftn(self.x3_fft)
assert_array_almost_equal(my_fftn(my_x),
self.x3_fft)
def test_ifftn_3d_axes(self):
test_axes = (0, -1)
my_ifftn = ifftn(self.x3_fft, axes=test_axes)
assert_array_almost_equal(my_ifftn(self.x3_fft),
np.fft.ifftn(self.x3_fft, axes=test_axes))
def test_ifftn_3d_mixed_and_short_shape(self):
test_shape = (2, 16)
my_ifftn = ifftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_ifftn(self.x3_fft),
np.fft.ifftn(self.x3_fft, s=test_shape))
def test_ifftn_3d_smaller_shape(self):
test_shape = (3, 3, 6)
my_ifftn = ifftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_ifftn(self.x3_fft),
np.fft.ifftn(self.x3_fft, s=test_shape))
def test_ifftn_3d_larger_shape(self):
test_shape = (4, 6, 14)
my_ifftn = ifftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_ifftn(self.x3_fft),
np.fft.ifftn(self.x3_fft, s=test_shape))
def test_ifftn_2d(self):
my_ifftn = ifftn(self.x2_fft)
assert_array_almost_equal(my_ifftn(self.x2_fft),
np.fft.ifftn(self.x2_fft))