def eb_modes(cmd_args):
#Plot setup
fig, ax = plt.subplots()
#Load in the shear map, compute E and B modes power spectrum
shear = ShearMap.load(
os.path.join(dataExtern(), "WLshear_z2.00_0001r.fits"))
l_edges = np.linspace(200., 50000., 50)
l, ee, bb, eb = shear.decompose(l_edges)
#Plot the power spectra and prediction from NICAEA
ax.plot(l, l * (l + 1) * ee / (2. * np.pi), label=r"$P^{EE}$", color="red")
cosmo = Nicaea(Om0=0.26, Ode0=0.74, w0=-1, sigma8=0.8)
ax.plot(l,
l * (l + 1) * cosmo.convergencePowerSpectrum(l, z=2.0) /
(2. * np.pi),
label=r"$P^{\kappa\kappa}{\rm (NICAEA)}$",
linestyle="--",
color="red")
ax.plot(l,
l * (l + 1) * bb / (2. * np.pi),
label=r"$P^{BB}$",
color="blue")
#Labels
ax.set_xscale("log")
ax.set_yscale("log")
ax.set_xlabel(r"$\ell$", fontsize=22)
ax.set_ylabel(r"$\ell(\ell+1)P_\ell/2\pi$", fontsize=22)
ax.legend(loc="upper left")
#Save
fig.savefig("eb_modes." + cmd_args.type)
shear_file_1 = fits.open(filename1)
angle = shear_file_1[0].header["ANGLE"]
gamma = shear_file_1[0].data.astype(np.float)
shear_file_1.close()
shear_file_2 = fits.open(filename2)
assert shear_file_2[0].header["ANGLE"] == angle
gamma = np.array((gamma, shear_file_2[0].data.astype(np.float)))
shear_file_2.close()
return angle * deg, gamma
test_map = ShearMap.load("Data/shear1.fit",
filename2="Data/shear2.fit",
format=two_file_loader)
test_map_conv = ConvergenceMap.load("Data/conv.fit")
l_edges = np.arange(200.0, 50000.0, 200.0)
def test_visualize1():
assert hasattr(test_map, "data")
assert hasattr(test_map, "side_angle")
assert test_map.data.shape[0] == 2
test_map.setAngularUnits(arcsec)
test_map.visualize(colorbar=True)
test_map.fig.tight_layout()
shear_file_1 = fits.open(filename1)
angle = shear_file_1[0].header["ANGLE"]
gamma = shear_file_1[0].data.astype(np.float)
shear_file_1.close()
shear_file_2 = fits.open(filename2)
assert shear_file_2[0].header["ANGLE"] == angle
gamma = np.array((gamma,shear_file_2[0].data.astype(np.float)))
shear_file_2.close()
return angle*deg,gamma
test_map = ShearMap.load("Data/shear1.fit",filename2="Data/shear2.fit",format=two_file_loader)
l_edges = np.arange(200.0,50000.0,200.0)
l,EE,BB,EB = test_map.decompose(l_edges,keep_fourier=True)
assert l.shape == EE.shape == BB.shape == EB.shape
fig,ax = plt.subplots()
ax.plot(l,l*(l+1)*EE/(2.0*np.pi),label=r"$P_{EE}$")
ax.plot(l,l*(l+1)*BB/(2.0*np.pi),label=r"$P_{BB}$")
ax.plot(l,l*(l+1)*np.abs(EB)/(2.0*np.pi),label=r"$\vert P_{EB}\vert$")
ax.set_xscale("log")
ax.set_yscale("log")
ax.set_xlabel(r"$l$")
