def test_cosmology_sane():
c = Cosmology(gauge='synchronous', verbose=True)
assert_allclose(c.Omega_cdm(0), c.Omega0_cdm)
assert_allclose(c.Omega_g(0), c.Omega0_g)
assert_allclose(c.Omega_b(0), c.Omega0_b)
assert_allclose(c.Omega_ncdm(0), c.Omega0_ncdm)
assert_allclose(c.Omega_ur(0), c.Omega0_ur)
assert_allclose(c.Omega_ncdm(0), c.Omega0_ncdm_tot)
assert_allclose(c.Omega_pncdm(0), c.Omega0_pncdm)
assert_allclose(c.Omega_m(0), c.Omega0_m)
assert_allclose(c.Omega_r(0), c.Omega0_r)
# total density in 10e10 Msun/h unit
assert_allclose(c.rho_tot(0), 27.754999)
# comoving distance to z=1.0 in Mpc/h unit.
assert_allclose(c.comoving_distance(1.0), 3396.157391 * c.h)
# conformal time in Mpc unit.
assert_allclose(c.tau(1.0), 3396.157391)
assert_allclose(c.efunc(0), 1.) # hubble in Mpc/h km/s unit
assert_allclose(c.efunc(0) - c.efunc(1 / 0.9999 - 1),
0.0001 * c.efunc_prime(0),
rtol=1e-3)
def test_cosmology_density():
c = Cosmology(gauge='synchronous')
z = [0, 1, 2, 5, 9, 99]
assert_allclose(c.rho_cdm(z), c.Omega_cdm(z) * c.rho_tot(z))
assert_allclose(c.rho_g(z), c.Omega_g(z) * c.rho_tot(z))
assert_allclose(c.rho_ncdm(z), c.Omega_ncdm(z) * c.rho_tot(z))
assert_allclose(c.rho_b(z), c.Omega_b(z) * c.rho_tot(z))
assert_allclose(c.rho_m(z), c.Omega_m(z) * c.rho_tot(z))
assert_allclose(c.rho_r(z), c.Omega_r(z) * c.rho_tot(z))
assert_allclose(c.rho_ur(z), c.Omega_ur(z) * c.rho_tot(z))
