def test_k2g_dim():
"""Test that conv2shear raises an error when k is not a 1D array."""
k = np.zeros((100, 100))
with np.testing.assert_raises_regex(Exception, '1D'):
transformations.conv2shear(k)
def test_k2g_mats_subset():
"""Test that conv2shear_mats produce equivalent output to conv2shear for a subset of indices."""
nside = 16
npix = hp.nside2npix(nside)
lmax = 32
ninds = 100
inds = np.random.choice(npix, ninds, replace=False)
A1, A2 = transformations.conv2shear_mats(nside, indices=inds, lmax=lmax)
np.testing.assert_almost_equal(A1.shape[0], ninds)
np.testing.assert_almost_equal(A1.shape[1], ninds)
np.testing.assert_almost_equal(A2.shape[0], ninds)
np.testing.assert_almost_equal(A2.shape[1], ninds)
k = np.zeros(npix)
k[inds] = np.random.standard_normal(ninds)
g1, g2 = transformations.conv2shear(k, lmax=lmax)
g1_lin = A1 @ k[inds]
g2_lin = A2 @ k[inds]
np.testing.assert_almost_equal(g1[inds], g1_lin)
np.testing.assert_almost_equal(g2[inds], g2_lin)
def test_k2g_zero():
"""Test that transformation of a zero convergence map is a zero shear map."""
nside = 16
npix = hp.nside2npix(nside)
k = np.zeros(npix)
g1, g2 = transformations.conv2shear(k)
np.testing.assert_almost_equal(g1, 0)
np.testing.assert_almost_equal(g2, 0)
def test_k2g2k_rand():
"""Test that transformation of k->g->k recovers the input convergence map."""
nside = 16
npix = hp.nside2npix(nside)
lmax = 32
k = np.random.standard_normal(npix)
k = hp.smoothing(k, lmax=lmax, verbose=False)
k = hp.remove_monopole(k)
k = hp.remove_dipole(k)
g1, g2 = transformations.conv2shear(k, lmax)
k_recov = transformations.shear2conv(g1, g2, lmax)
np.testing.assert_almost_equal(k, k_recov, decimal=3)
def test_k2g_mats_full():
"""Test that conv2shear_mats produce equivalent output to conv2shear."""
nside = 16
npix = hp.nside2npix(nside)
lmax = 32
A1, A2 = transformations.conv2shear_mats(nside, lmax=lmax)
k = np.random.standard_normal(npix)
g1, g2 = transformations.conv2shear(k, lmax=lmax)
g1_lin = A1 @ k
g2_lin = A2 @ k
np.testing.assert_almost_equal(g1, g1_lin)
np.testing.assert_almost_equal(g2, g2_lin)