def test_ihfft2(self): x = random((30, 20)) expect = fft.ifft2(x)[:, :11] assert_array_almost_equal(expect, fft.ihfft2(x)) assert_array_almost_equal(expect, fft.ihfft2(x, norm="backward")) assert_array_almost_equal(expect * np.sqrt(30 * 20), fft.ihfft2(x, norm="ortho")) assert_array_almost_equal(expect * (30 * 20), fft.ihfft2(x, norm="forward"))
def test_ihfft2(self): x = random((30, 20)) assert_array_almost_equal(fft.ifft2(x)[:, :11], fft.ihfft2(x)) assert_array_almost_equal(fft.ihfft2(x) * np.sqrt(30 * 20), fft.ihfft2(x, norm="ortho"))
def test_hfft2(self): x = random((30, 20)) assert_array_almost_equal(x, fft.hfft2(fft.ihfft2(x))) assert_array_almost_equal( x, fft.hfft2(fft.ihfft2(x, norm="ortho"), norm="ortho"))
def test_hfft2(self): x = random((30, 20)) assert_array_almost_equal(x, fft.hfft2(fft.ihfft2(x))) for norm in ["backward", "ortho", "forward"]: assert_array_almost_equal( x, fft.hfft2(fft.ihfft2(x, norm=norm), norm=norm))