def test_cosmology_get_pk(): c = Cosmology() p = c.get_pk(z=0, k=0.1) p1 = c.Spectra.get_pk(z=0, k=0.1) # ensure the dro did use get_pk of Spectra rather than that from Primordial assert_allclose(p, p1)
def test_cosmology_vect(): c = Cosmology(gauge='synchronous') assert_allclose(c.Omega_cdm([0]), c.Omega0_cdm) assert_array_equal(c.Omega_cdm([]).shape, [0]) assert_array_equal(c.Omega_cdm([0]).shape, [1]) assert_array_equal(c.Omega_cdm([[0]]).shape, [1, 1]) assert_array_equal(c.rho_k([[0]]).shape, [1, 1]) k, z = numpy.meshgrid([0, 1], [0.01, 0.05, 0.1, 0.5], sparse=True, indexing='ij') pk = c.get_pk(z=z, k=k) assert_array_equal(pk.shape, [2, 4])