def test_irfftn_3d_mixed_and_short_shape(self):
test_shape = (6, 12)
my_irfftn = irfftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_irfftn(self.x3_fft),
np.fft.irfftn(self.x3_fft, s=test_shape))
def test_irfftn_3d_axes(self):
test_axes = (0, -1)
my_irfftn = irfftn(self.x3_fft, axes=test_axes)
assert_array_almost_equal(my_irfftn(self.x3_fft),
np.fft.irfftn(self.x3_fft, axes=test_axes))
def test_irfftn_3d_larger_shape(self):
test_shape = (4, 6, 32)
my_irfftn = irfftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_irfftn(self.x3_fft),
np.fft.irfftn(self.x3_fft, s=test_shape))
def test_irfftn_3d_smaller_shape(self):
test_shape = (3, 3, 12)
my_irfftn = irfftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_irfftn(self.x3_fft),
np.fft.irfftn(self.x3_fft, s=test_shape))
def test_irfftn_pair(self):
my_irfftn, my_rfftn = irfftn(self.x3_fft,
fft_pair=True)
my_x = my_irfftn(self.x3_fft)
assert_array_almost_equal(my_rfftn(my_x),
self.x3_fft)
def test_irfftn_2d(self):
my_irfftn = irfftn(self.x2_fft)
assert_array_almost_equal(my_irfftn(self.x2_fft),
np.fft.irfftn(self.x2_fft))
def test_irfftn_3d_axes(self):
test_axes = (0, -1)
my_irfftn = irfftn(self.x3_fft, axes=test_axes)
assert_array_almost_equal(my_irfftn(self.x3_fft),
np.fft.irfftn(self.x3_fft, axes=test_axes))
def test_irfftn_3d_mixed_and_short_shape(self):
test_shape = (6, 12)
my_irfftn = irfftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_irfftn(self.x3_fft),
np.fft.irfftn(self.x3_fft, s=test_shape))
def test_irfftn_3d_smaller_shape(self):
test_shape = (3, 3, 12)
my_irfftn = irfftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_irfftn(self.x3_fft),
np.fft.irfftn(self.x3_fft, s=test_shape))
def test_irfftn_3d_larger_shape(self):
test_shape = (4, 6, 32)
my_irfftn = irfftn(self.x3_fft, shape=test_shape)
assert_array_almost_equal(my_irfftn(self.x3_fft),
np.fft.irfftn(self.x3_fft, s=test_shape))
def test_irfftn_2d(self):
my_irfftn = irfftn(self.x2_fft)
assert_array_almost_equal(my_irfftn(self.x2_fft),
np.fft.irfftn(self.x2_fft))