def check_neutrinos(): """ Check that neutrino-related functions can be run. """ z = 0. z_arr = np.linspace(0., 2., 10) a = 1. a_arr = 1. / (1. + z_arr) a_lst = [_a for _a in a_arr] T_CMB = 2.725 N_nu_mass = 3 mnu = [0.02, 0.02, 0.02] # Omeganuh2 assert_(all_finite(ccl.Omeganuh2(a, mnu, T_CMB))) assert_(all_finite(ccl.Omeganuh2(a_lst, mnu, T_CMB))) assert_(all_finite(ccl.Omeganuh2(a_arr, mnu, T_CMB))) OmNuh2 = 0.01 # Omeganuh2_to_Mnu assert_(all_finite(ccl.nu_masses(OmNuh2, 'normal', T_CMB))) assert_(all_finite(ccl.nu_masses(OmNuh2, 'inverted', T_CMB))) assert_(all_finite(ccl.nu_masses(OmNuh2, 'equal', T_CMB))) assert_(all_finite(ccl.nu_masses(OmNuh2, 'sum', T_CMB))) # Check that the right exceptions are raised assert_raises(ValueError, ccl.Cosmology, Omega_c=0.27, Omega_b=0.045, h=0.67, A_s=1e-10, n_s=0.96, m_nu=[0.1, 0.2, 0.3, 0.4]) assert_raises(ValueError, ccl.Cosmology, Omega_c=0.27, Omega_b=0.045, h=0.67, A_s=1e-10, n_s=0.96, m_nu=[0.1, 0.2, 0.3], mnu_type="sum") assert_raises(ValueError, ccl.Cosmology, Omega_c=0.27, Omega_b=0.045, h=0.67, A_s=1e-10, n_s=0.96, m_nu=42)
def test_omnuh2_smoke(a, m): om = ccl.Omeganuh2(a, m) assert np.all(np.isfinite(om)) assert np.shape(om) == np.shape(a)
def test_nu_mass_consistency(a, split): m = ccl.nu_masses(0.1, split) assert np.allclose(ccl.Omeganuh2(a, m), 0.1, rtol=0, atol=1e-4)