def test_fftn_pair_cropped(self): test_axes = (0, -1) test_shape = (4, 16) my_fftn, my_ifftn = fftn(self.x3, shape=test_shape, axes=test_axes, fft_pair=True, crop_ifft=True) x_fft = my_fftn(self.x3) assert_array_almost_equal(my_ifftn(x_fft), self.x3)
def test_fftn_pair(self): test_axes = (0, -1) test_shape = (4, 16) my_fftn, my_ifftn = fftn(self.x3, shape=test_shape, axes=test_axes, fft_pair=True) x_fft = my_fftn(self.x3) assert_array_almost_equal(my_ifftn(x_fft), pad_array(self.x3, (4, 3, 16)))
def test_fftn_3d_axes(self): test_axes = (0, -1) my_fftn = fftn(self.x3, axes=test_axes) assert_array_almost_equal(my_fftn(self.x3), np.fft.fftn(self.x3, axes=test_axes))
def test_fftn_3d_smaller_shape(self): test_shape = (3, 3, 6) my_fftn = fftn(self.x3, shape=test_shape) assert_array_almost_equal(my_fftn(self.x3), np.fft.fftn(self.x3, s=test_shape))
def test_fftn_3d_mixed_and_short_shape(self): test_shape = (2, 16) my_fftn = fftn(self.x3, shape=test_shape) assert_array_almost_equal(my_fftn(self.x3), np.fft.fftn(self.x3, s=test_shape))
def test_fftn_2d(self): my_fftn = fftn(self.x2) assert_array_almost_equal(my_fftn(self.x2), np.fft.fftn(self.x2))
def test_fftn_3d_larger_shape(self): test_shape = (4, 6, 14) my_fftn = fftn(self.x3, shape=test_shape) assert_array_almost_equal(my_fftn(self.x3), np.fft.fftn(self.x3, s=test_shape))