@@ -22,27 +22,6 @@ def test_healpix_fft_jax_numpy_consistency(rng, L, nside, reality):
2222 )
2323
2424
25- @pytest .mark .parametrize ("L" , (32 , 64 ))
26- @pytest .mark .parametrize ("nside" , (4 , 8 , 16 ))
27- @pytest .mark .parametrize ("reality" , (True , False ))
28- def test_healpix_fft_ifft_numpy_consistency (rng , L , nside , reality ):
29- f = rng .standard_normal (size = 12 * nside ** 2 )
30- assert np .allclose (
31- f ,
32- healpix_ifft_numpy (healpix_fft_numpy (f , L , nside , reality ), L , nside , reality ),
33- )
34-
35-
36- @pytest .mark .parametrize ("L" , (32 , 64 ))
37- @pytest .mark .parametrize ("nside" , (4 , 8 , 16 ))
38- @pytest .mark .parametrize ("reality" , (True , False ))
39- def test_healpix_fft_ifft_jax_consistency (rng , L , nside , reality ):
40- f = rng .standard_normal (size = 12 * nside ** 2 )
41- assert np .allclose (
42- f , healpix_ifft_jax (healpix_fft_jax (f , L , nside , reality ), L , nside , reality )
43- )
44-
45-
4625@pytest .mark .parametrize ("L" , (32 , 64 ))
4726@pytest .mark .parametrize ("nside" , (4 , 8 , 16 ))
4827@pytest .mark .parametrize ("reality" , (True , False ))
0 commit comments