def test_hpxgeom_contains(nside, nested, frame, region, axes):
geom = HpxGeom(nside, nested, frame, region=region, axes=axes)
coords = geom.get_coord(flat=True)
assert_allclose(geom.contains(coords), np.ones_like(coords[0], dtype=bool))
if axes is not None:
coords = [c[0]
for c in coords[:2]] + [ax.edges[-1] + 1.0 for ax in axes]
assert_allclose(geom.contains(coords), np.zeros((1, ), dtype=bool))
if geom.region is not None:
coords = [0.0, 0.0] + [ax.center[0] for ax in geom.axes]
assert_allclose(geom.contains(coords), np.zeros((1, ), dtype=bool))
def test_hpxgeom_upsample(nside, nested, frame, region, axes):
# NESTED
geom = HpxGeom(nside, True, frame, region=region, axes=axes)
geom_up = geom.upsample(2)
assert_allclose(2 * geom.nside, geom_up.nside)
assert_allclose(4 * geom.npix, geom_up.npix)
coords = geom_up.get_coord(flat=True)
assert np.all(geom.contains(coords))
# RING
geom = HpxGeom(nside, False, frame, region=region, axes=axes)
geom_up = geom.upsample(2)
assert_allclose(2 * geom.nside, geom_up.nside)
assert_allclose(4 * geom.npix, geom_up.npix)
coords = geom_up.get_coord(flat=True)
assert np.all(geom.contains(coords))
