def test_wcsndmap_upsample_axis(): axis = MapAxis.from_edges([1, 2, 3, 4], name="test") geom = WcsGeom.create(npix=(4, 4), axes=[axis]) m = WcsNDMap(geom, unit="m2") m.data += 1 m2 = m.upsample(2, preserve_counts=True, axis_name="test") assert m2.data.shape == (6, 4, 4) assert_allclose(m.data.sum(), m2.data.sum())
def test_wcsndmap_upsample(npix, binsz, frame, proj, skydir, axes): geom = WcsGeom.create(npix=npix, binsz=binsz, proj=proj, frame=frame, axes=axes) m = WcsNDMap(geom, unit="m2") m2 = m.upsample(2, preserve_counts=True) assert_allclose(np.nansum(m.data), np.nansum(m2.data)) assert m.unit == m2.unit