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_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])
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])
