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_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_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_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_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_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))