def test_hpxmap_downsample(nside, nested, coordsys, region, axes): m = HpxNDMap( HpxGeom(nside=nside, nest=nested, coordsys=coordsys, region=region, axes=axes), unit="m2", ) m.set_by_pix(m.geom.get_idx(flat=True), 1.0) m_down = m.downsample(2, preserve_counts=True) assert_allclose(np.nansum(m.data), np.nansum(m_down.data)) assert m.unit == m_down.unit
def test_hpxmap_upsample(nside, nested, frame, region, axes): m = HpxNDMap( HpxGeom(nside=nside, nest=nested, frame=frame, region=region, axes=axes), unit="m2", ) m.set_by_pix(m.geom.get_idx(flat=True), 1.0) m_up = m.upsample(2, preserve_counts=True) assert_allclose(np.nansum(m.data), np.nansum(m_up.data)) m_up = m.upsample(2, preserve_counts=False) assert_allclose(4.0 * np.nansum(m.data), np.nansum(m_up.data)) assert m.unit == m_up.unit
def test_hpxmap_pad(nside, nested, coordsys, region, axes): m = HpxNDMap( HpxGeom(nside=nside, nest=nested, coordsys=coordsys, region=region, axes=axes)) m.set_by_pix(m.geom.get_idx(flat=True), 1.0) cval = 2.2 m_pad = m.pad(1, mode="constant", cval=cval) coords_pad = m_pad.geom.get_coord(flat=True) msk = m.geom.contains(coords_pad) coords_out = tuple([c[~msk] for c in coords_pad]) assert_allclose(m_pad.get_by_coord(coords_out), cval * np.ones_like(coords_out[0])) coords_in = tuple([c[msk] for c in coords_pad]) assert_allclose(m_pad.get_by_coord(coords_in), np.ones_like(coords_in[0]))