def test_hpxgeom_init_with_pix(nside, nested, frame, region, axes): geom = HpxGeom(nside, nested, frame, region=region, axes=axes) idx0 = geom.get_idx(flat=True) idx1 = tuple([t[::10] for t in idx0]) geom = HpxGeom(nside, nested, frame, region=idx0, axes=axes) assert_allclose(idx0, geom.get_idx(flat=True)) assert_allclose(len(idx0[0]), np.sum(geom.npix)) geom = HpxGeom(nside, nested, frame, region=idx1, axes=axes) assert_allclose(idx1, geom.get_idx(flat=True)) assert_allclose(len(idx1[0]), np.sum(geom.npix))
def test_hpxgeom_get_pix(nside, nested, frame, region, axes): geom = HpxGeom(nside, nested, frame, region=region, axes=axes) idx = geom.get_idx(local=False, flat=True) idx_local = geom.get_idx(local=True, flat=True) assert_allclose(idx, geom.local_to_global(idx_local)) if axes is not None: idx_img = geom.get_idx(local=False, idx=tuple([1] * len(axes)), flat=True) idx_img_local = geom.get_idx(local=True, idx=tuple([1] * len(axes)), flat=True) assert_allclose(idx_img, geom.local_to_global(idx_img_local))