ccl_test_power Namespace Reference

Functions
def	all_finite (vals)
def	calc_power_spectrum (Omega_v, w0, wa, transfer_fn, matter_power, linear, raise_errors)
def	loop_over_params (transfer_fn, matter_power, lin, raise_errs)
def	test_power_spectrum_linear ()
def	test_power_spectrum_linear_slow ()
def	test_power_spectrum_halofit ()
def	test_power_spectrum_halofit_slow ()
def	test_power_spectrum_emu ()
def	test_nonlin_power_spectrum_linear ()
def	test_nonlin_power_spectrum_linear_slow ()
def	test_nonlin_power_spectrum_halofit ()
def	test_nonlin_power_spectrum_halofit_slow ()
def	test_nonlin_power_spectrum_emu ()
def	test_raise_error_EH_bbks_lin ()
def	test_raise_error_EH_bbks_halofit ()
def	test_raise_error_EH_bbks_nonlin_linear ()
def	test_raise_error_emu ()
def	test_raise_error_emu_nonlin ()

Variables
float	Omega_c = 0.25
float	Omega_b = 0.045
float	Neff = 3.046
float	mnu_sum = 0.06
list	mnu_list = [0.02, 0.02, 0.02]
float	h = 0.7
float	sigma8 = 0.83
float	n_s = 0.96
	Omega_v_vals = np.array([0.7, 0.7, 0.7, 0.65, 0.75])
	w0_vals = np.array([-1.0, -0.9, -0.9, -0.9, -0.9])
	wa_vals = np.array([0.0, 0.0, 0.1, 0.1, 0.1])
list	transfer_fns = ['boltzmann_class', 'eisenstein_hu', 'emulator',]
	argv

Function Documentation

def ccl_test_power.all_finite

(

vals

)

Returns True if all elements are finite (i.e. not NaN or inf).

Definition at line 28 of file ccl_test_power.py.

Referenced by calc_power_spectrum().

def all_finite(vals):
    """
    Returns True if all elements are finite (i.e. not NaN or inf).
    """
    return np.all( np.isfinite(vals) )

def ccl_test_power.calc_power_spectrum	(		Omega_v,
			w0,
			wa,
			transfer_fn,
			matter_power,
			linear,
			raise_errors
	)

Calculate linear and nonlinear power spectrum for a given set of parameters
and choices of transfer function and matter power spectrum.

Definition at line 35 of file ccl_test_power.py.

References all_finite().

Referenced by loop_over_params().

def calc_power_spectrum(Omega_v, w0, wa, transfer_fn, matter_power, linear, raise_errors):
    """
    Calculate linear and nonlinear power spectrum for a given set of parameters
    and choices of transfer function and matter power spectrum.
    """
    k = np.logspace(-5., 1., 300)
    a = np.logspace(np.log10(0.51), 0., 5) # Emulator only works at z<2

    # Set Omega_K in a consistent way
    Omega_k = 1.0 - Omega_c - Omega_b - Omega_v

    if (raise_errors == False):
        if (transfer_fn == 'eisenstein_hu' or transfer_fn == 'bbks'):
            mnu = 0. # The bbks and E-H P(k) are not defined for massive neutrinos.
        elif (transfer_fn == 'emulator' and matter_power=='emu'):
            mnu = mnu_list # For the emulator, we must have 3 equal masses
        else:
            mnu = mnu_sum
    elif (raise_errors==True):
        if (transfer_fn =='eisenstein_hu' or transfer_fn =='bbks'):
            mnu = mnu_sum #Use massive neutrinos to deliberately raise an error
        elif (transfer_fn == 'emulator' and matter_power == 'emu'):
            mnu = mnu_sum #Use a sum instead of an equal list to deliberately raise an error.
        else:
            raise(ValueError, "Transfer function %s with matter power spectrum method %s has no case for which to test errors are raised." % (transfer_fn, matter_power))

    # Create a new Cosmology object
    cosmo = ccl.Cosmology(Omega_c=Omega_c, Omega_b=Omega_b,
                       h=h, sigma8=sigma8, n_s=n_s, Omega_k=Omega_k,
                       w0=w0, wa=wa, transfer_function=transfer_fn,
                       matter_power_spectrum=matter_power,
                       Neff = Neff, m_nu = mnu)

    # Calculate linear and nonlinear power spectra for each scale factor, a
    for _a in a:
        if linear:
            if raise_errors==False:
                pk_lin = ccl.linear_matter_power(cosmo, k, _a)
                assert_(all_finite(pk_lin))
            else:
                assert_raises(CCLError, ccl.linear_matter_power, cosmo, k, _a)
        else:
            if raise_errors==False:
                pk_nl = ccl.nonlin_matter_power(cosmo, k, _a)
                assert_(all_finite(pk_nl))
            else:
                assert_raises(CCLError,ccl.nonlin_matter_power, cosmo, k, _a)

def ccl_test_power.loop_over_params	(		transfer_fn,
			matter_power,
			lin,
			raise_errs
	)

Call the power spectrum testing function for each of a set of parameters.

Definition at line 83 of file ccl_test_power.py.

References calc_power_spectrum().

Referenced by test_nonlin_power_spectrum_emu(), test_nonlin_power_spectrum_halofit(), test_nonlin_power_spectrum_halofit_slow(), test_nonlin_power_spectrum_linear(), test_nonlin_power_spectrum_linear_slow(), test_power_spectrum_emu(), test_power_spectrum_halofit(), test_power_spectrum_halofit_slow(), test_power_spectrum_linear(), test_power_spectrum_linear_slow(), test_raise_error_EH_bbks_halofit(), test_raise_error_EH_bbks_lin(), test_raise_error_EH_bbks_nonlin_linear(), test_raise_error_emu(), and test_raise_error_emu_nonlin().

def loop_over_params(transfer_fn, matter_power, lin, raise_errs):
    """
    Call the power spectrum testing function for each of a set of parameters.
    """
    print(">>> (%s; %s)" % (transfer_fn, matter_power))

    # Loop over parameters
    for i in [0,2]: #w0_vals.size):
        calc_power_spectrum(Omega_v_vals[i], w0_vals[i], wa_vals[i],
                            transfer_fn=transfer_fn, matter_power=matter_power,
                            linear=lin, raise_errors = raise_errs)

def ccl_test_power.test_nonlin_power_spectrum_emu

(

)

Definition at line 136 of file ccl_test_power.py.

References loop_over_params().

def test_nonlin_power_spectrum_emu():
    transfer_fns = ['emulator',]
    for tfn in transfer_fns: loop_over_params(tfn, 'emu', lin=False, raise_errs = False)

@decorators.slow

def ccl_test_power.test_nonlin_power_spectrum_halofit

(

)

Definition at line 126 of file ccl_test_power.py.

References loop_over_params().

def test_nonlin_power_spectrum_halofit():
    for tfn in ['eisenstein_hu', 'bbks']:
        loop_over_params(tfn, 'halofit', lin=False, raise_errs = False)

@decorators.slow

def ccl_test_power.test_nonlin_power_spectrum_halofit_slow

(

)

Definition at line 131 of file ccl_test_power.py.

References loop_over_params().

def test_nonlin_power_spectrum_halofit_slow():
    for tfn in ['boltzmann',]:
        loop_over_params(tfn, 'halofit', lin=False, raise_errs = False)

@decorators.slow

def ccl_test_power.test_nonlin_power_spectrum_linear

(

)

Definition at line 117 of file ccl_test_power.py.

References loop_over_params().

def test_nonlin_power_spectrum_linear():
    for tfn in ['eisenstein_hu', 'bbks']:
        loop_over_params(tfn, 'linear', lin=False, raise_errs = False)

@decorators.slow

def ccl_test_power.test_nonlin_power_spectrum_linear_slow

(

)

Definition at line 122 of file ccl_test_power.py.

References loop_over_params().

def test_nonlin_power_spectrum_linear_slow():
    for tfn in ['boltzmann',]:
        loop_over_params(tfn, 'linear', lin=False, raise_errs = False)

def ccl_test_power.test_power_spectrum_emu

(

)

Definition at line 114 of file ccl_test_power.py.

References loop_over_params().

def test_power_spectrum_emu():
    for tfn in ['emulator',]: loop_over_params(tfn, 'emu', lin=True, raise_errs = False)

def ccl_test_power.test_power_spectrum_halofit

(

)

Definition at line 104 of file ccl_test_power.py.

References loop_over_params().

def test_power_spectrum_halofit():
    for tfn in ['eisenstein_hu', 'bbks',]:
        loop_over_params(tfn, 'halofit', lin=True, raise_errs = False)

@decorators.slow

def ccl_test_power.test_power_spectrum_halofit_slow

(

)

Definition at line 109 of file ccl_test_power.py.

References loop_over_params().

def test_power_spectrum_halofit_slow():
    for tfn in ['boltzmann',]:
        loop_over_params(tfn, 'halofit', lin=True, raise_errs = False)

@decorators.slow

def ccl_test_power.test_power_spectrum_linear

(

)

Definition at line 95 of file ccl_test_power.py.

References loop_over_params().

def test_power_spectrum_linear():
    for tfn in ['bbks', 'eisenstein_hu']:
        loop_over_params(tfn, 'linear', lin=True, raise_errs = False)

@decorators.slow

def ccl_test_power.test_power_spectrum_linear_slow

(

)

Definition at line 100 of file ccl_test_power.py.

References loop_over_params().

def test_power_spectrum_linear_slow():
    for tfn in ['boltzmann',]:
        loop_over_params(tfn, 'linear', lin=True, raise_errs = False)

def ccl_test_power.test_raise_error_EH_bbks_halofit

(

)

Definition at line 146 of file ccl_test_power.py.

References loop_over_params().

def test_raise_error_EH_bbks_halofit():
    for tfn in ['bbks']:
        loop_over_params(tfn, 'halofit', lin=False, raise_errs=True)

@decorators.slow

def ccl_test_power.test_raise_error_EH_bbks_lin

(

)

Definition at line 141 of file ccl_test_power.py.

References loop_over_params().

def test_raise_error_EH_bbks_lin():
    for tfn in ['eisenstein_hu']:
        loop_over_params(tfn, 'linear', lin=True, raise_errs=True)

@decorators.slow

def ccl_test_power.test_raise_error_EH_bbks_nonlin_linear

(

)

Definition at line 151 of file ccl_test_power.py.

References loop_over_params().

def test_raise_error_EH_bbks_nonlin_linear():
    for tfn in ['bbks']:
        loop_over_params(tfn, 'linear', lin=False, raise_errs=True)

@decorators.slow

def ccl_test_power.test_raise_error_emu

(

)

Definition at line 156 of file ccl_test_power.py.

References loop_over_params().

def test_raise_error_emu():
    transfer_fns = ['emulator',]
    for tfn in transfer_fns: loop_over_params(tfn, 'emu', lin=True, raise_errs = True)

@decorators.slow

def ccl_test_power.test_raise_error_emu_nonlin

(

)

Definition at line 161 of file ccl_test_power.py.

References loop_over_params().

def test_raise_error_emu_nonlin():
    transfer_fns = ['emulator',]
    for tfn in transfer_fns: loop_over_params(tfn, 'emu', lin=False, raise_errs = True)

Variable Documentation

ccl_test_power.argv

Definition at line 166 of file ccl_test_power.py.

Referenced by TEST_CASE().

float ccl_test_power.h = 0.7

Definition at line 16 of file ccl_test_power.py.

list ccl_test_power.mnu_list = [0.02, 0.02, 0.02]

Definition at line 15 of file ccl_test_power.py.

float ccl_test_power.mnu_sum = 0.06

Definition at line 14 of file ccl_test_power.py.

float ccl_test_power.n_s = 0.96

Definition at line 18 of file ccl_test_power.py.

float ccl_test_power.Neff = 3.046

Definition at line 13 of file ccl_test_power.py.

float ccl_test_power.Omega_b = 0.045

Definition at line 12 of file ccl_test_power.py.

float ccl_test_power.Omega_c = 0.25

Definition at line 11 of file ccl_test_power.py.

ccl_test_power.Omega_v_vals = np.array([0.7, 0.7, 0.7, 0.65, 0.75])

Definition at line 21 of file ccl_test_power.py.

float ccl_test_power.sigma8 = 0.83

Definition at line 17 of file ccl_test_power.py.

Referenced by __attribute__(), ccl_cosmology_compute_power_bbks(), ccl_cosmology_compute_power_eh(), ccl_get_class_As(), ccl_parameters_read_yaml(), ccl_parameters_write_yaml(), and write_Pk_CLASS().

list ccl_test_power.transfer_fns = ['boltzmann_class', 'eisenstein_hu', 'emulator',]

Definition at line 26 of file ccl_test_power.py.

ccl_test_power.w0_vals = np.array([-1.0, -0.9, -0.9, -0.9, -0.9])

Definition at line 22 of file ccl_test_power.py.

ccl_test_power.wa_vals = np.array([0.0, 0.0, 0.1, 0.1, 0.1])

Definition at line 23 of file ccl_test_power.py.