ccl_test_swig_interface.py

Go to the documentation of this file.

import copy
import numpy as np
from numpy.testing import assert_raises, run_module_suite
import pyccl
from pyccl import ccllib
from pyccl import CCLError

PYCOSMO = pyccl.Cosmology(
    Omega_c=0.27, Omega_b=0.045, h=0.67, A_s=1e-10, n_s=0.96)
COSMO = PYCOSMO.cosmo


def test_swig_tracer():
    z = np.linspace(0, 1, 10)
    bias_ia = z * 2
    f_red = z / 1
    dNdz = z * 8

    assert_raises(
        CCLError,
        pyccl.WeakLensingTracer,
        PYCOSMO,
        dndz=(z, dNdz[0:2]),
        ia_bias=(z, bias_ia),
        red_frac=(z, f_red))

    assert_raises(
        CCLError,
        pyccl.WeakLensingTracer,
        PYCOSMO,
        dndz=(z, dNdz),
        ia_bias=(z, bias_ia[0:2]),
        red_frac=(z, f_red))

    assert_raises(
        CCLError,
        pyccl.WeakLensingTracer,
        PYCOSMO,
        dndz=(z, dNdz),
        ia_bias=(z, bias_ia),
        red_frac=(z, f_red[0:2]))


def test_swig_background():
    status = 0
    for func in [
            ccllib.growth_factor_vec,
            ccllib.growth_factor_unnorm_vec,
            ccllib.growth_rate_vec,
            ccllib.comoving_radial_distance_vec,
            ccllib.comoving_angular_distance_vec,
            ccllib.h_over_h0_vec,
            ccllib.luminosity_distance_vec,
            ccllib.distance_modulus_vec]:
        assert_raises(
            CCLError,
            func,
            COSMO,
            [0.0, 1.0],
            1,
            status)

    assert_raises(
        CCLError,
        ccllib.omega_x_vec,
        COSMO,
        0,
        [0.0, 1.0],
        1,
        status)

    assert_raises(
        CCLError,
        ccllib.rho_x_vec,
        COSMO,
        0,
        0,
        [0.0, 1.0],
        1,
        status)

    assert_raises(
        CCLError,
        ccllib.scale_factor_of_chi_vec,
        COSMO,
        [0.0, 1.0],
        1,
        status)


def test_swig_cls():
    status = 0
    base_args = [
        [0.0, 1.0],
        [0.0, 1.0],
        [0.0, 1.0, 2.0],
        [0.0, 1.0, 2.0],
        [0.0, 1.0, 2.0, 3.0],
        [0.0, 1.0, 2.0, 3.0],
        [0.0, 1.0, 2.0, 3.0, 4.0],
        [0.0, 1.0, 2.0, 3.0, 4.0],
        [0.0, 1.0, 2.0, 3.0, 4.0, 5.0],
        [0.0, 1.0, 2.0, 3.0, 4.0, 5.0],
        1.0,
        status]

    for i in range(5):
        args = copy.deepcopy(base_args)
        args[i*2] = [1.0] * 8
        assert_raises(
            CCLError,
            ccllib.cl_tracer_new_wrapper,
            COSMO, 0,
            0, 0, 0,
            *args)

    assert_raises(
        CCLError,
        ccllib.angular_cl_vec,
        COSMO,
        None, None,
        1, 1, 1,
        0,
        "none",
        status)

    assert_raises(
        CCLError,
        ccllib.clt_fa_vec,
        COSMO, None, 0,
        [0, 1, 2],
        2,
        status)

def test_swig_core():
    status = 0
    assert_raises(
        CCLError,
        ccllib.parameters_create_nu_vec,
        0.25, 0.05, 0.0, 3.0, -1.0, 0.0, 0.7, 2e-9, 0.95, 1, 0.0, 0.0,
        [1.0, 2.0],
        [0.0, 0.3, 0.5],
        0,
        [0.02, 0.01, 0.2],
        status)


def test_swig_correlation():
    status = 0
    assert_raises(
        CCLError,
        ccllib.correlation_vec,
        COSMO,
        [1, 2],
        [1, 2, 3],
        [0, 1, 2, 3],
        0, 0,
        4,
        status)

    assert_raises(
        CCLError,
        ccllib.correlation_vec,
        COSMO,
        [1, 2, 3],
        [1, 2, 3],
        [0, 1, 2, 3],
        0, 0,
        5,
        status)

    assert_raises(
        CCLError,
        ccllib.correlation_3d_vec,
        COSMO,
        1.0,
        [1, 2, 3],
        4,
        status)


def test_swig_halomod():
    status = 0
    for func in [ccllib.onehalo_matter_power_vec,
                 ccllib.twohalo_matter_power_vec,
                 ccllib.halomodel_matter_power_vec]:
        assert_raises(
            CCLError,
            func,
            COSMO,
            1.0,
            [0.1, 1.0],
            9,
            status)

    assert_raises(
        CCLError,
        ccllib.halo_concentration_vec,
        COSMO,
        1.0,
        200.0,
        [1e13, 1e14],
        8,
        status)


def test_swig_redshifts():
    status = 0

    assert_raises(
        CCLError,
        ccllib.dNdz_tomog_vec,
        0,
        0.0, 1.0,
        None,
        [0.0, 1.0],
        7,
        status)


def test_swig_massfunc():
    status = 0
    for func in [ccllib.massfunc_vec, ccllib.halo_bias_vec]:
        assert_raises(
            CCLError,
            func,
            COSMO,
            1.0, 200.0,
            [1e13, 1e14],
            4,
            status)

    assert_raises(
        CCLError,
        ccllib.massfunc_m2r_vec,
        COSMO,
        [1e13, 1e14],
        4,
        status)

    assert_raises(
        CCLError,
        ccllib.sigmaM_vec,
        COSMO,
        1.0,
        [1e13, 1e14],
        4,
        status)


def test_swig_neurtinos():
    status = 0
    assert_raises(
        CCLError,
        ccllib.Omeganuh2_vec,
        3, 2.7,
        [0.0, 1.0],
        [0.05, 0.1, 0.2],
        4,
        status)

    assert_raises(
        CCLError,
        ccllib.Omeganuh2_vec,
        3, 2.7,
        [0.0, 1.0],
        [0.1, 0.2],
        2,
        status)


def test_swig_power():
    status = 0
    for func in [ccllib.linear_matter_power_vec,
                 ccllib.nonlin_matter_power_vec]:
        assert_raises(
            CCLError,
            func,
            COSMO,
            1.0,
            [1.0, 2.0],
            3,
            status)

    for func in [ccllib.sigmaR_vec,
                 ccllib.sigmaV_vec]:
        assert_raises(
            CCLError,
            func,
            COSMO,
            1.0,
            [1.0, 2.0],
            3,
            status)

if __name__ == '__main__':
    run_module_suite()