def test_get_superpixels(nside_subpix, nside_superpix, nest):
import healpy as hp
npix = 12 * nside_subpix**2
subpix = np.arange(npix)
ang_subpix = hp.pix2ang(nside_subpix, subpix, nest=nest)
superpix = get_superpixels(subpix, nside_subpix, nside_superpix, nest=nest)
pix1 = hp.ang2pix(nside_superpix, *ang_subpix, nest=nest)
assert_allclose(superpix, pix1)
subpix = subpix.reshape((12, -1))
if not np.isscalar(nside_subpix):
nside_subpix = nside_subpix.reshape((12, -1))
if not np.isscalar(nside_superpix):
nside_superpix = nside_superpix.reshape((12, -1))
ang_subpix = hp.pix2ang(nside_subpix, subpix, nest=nest)
superpix = get_superpixels(subpix, nside_subpix, nside_superpix, nest=nest)
pix1 = hp.ang2pix(nside_superpix, *ang_subpix, nest=nest)
assert_allclose(superpix, pix1)
def test_get_subpixels(nside_superpix, nside_subpix, nest):
import healpy as hp
npix = 12 * nside_superpix**2
superpix = np.arange(npix)
subpix = get_subpixels(superpix, nside_superpix, nside_subpix, nest=nest)
ang1 = hp.pix2ang(nside_subpix, subpix, nest=nest)
pix1 = hp.ang2pix(nside_superpix, *ang1, nest=nest)
assert np.all(superpix[..., None] == pix1)
superpix = superpix.reshape((12, -1))
subpix = get_subpixels(superpix, nside_superpix, nside_subpix, nest=nest)
ang1 = hp.pix2ang(nside_subpix, subpix, nest=nest)
pix1 = hp.ang2pix(nside_superpix, *ang1, nest=nest)
assert np.all(superpix[..., None] == pix1)
pix1 = get_superpixels(subpix, nside_subpix, nside_superpix, nest=nest)
assert np.all(superpix[..., None] == pix1)