import cosmology

import correlation

import defaults

import halo

import halo_trispectrum

import hod

import kernel

import mass_function

import perturbation_spectra

import numpy

import unittest

deg_to_rad = numpy.pi/180.0

### In order for the unittests to work correctly, these are the assumed

### precision values of the code.

defaults.default_precision = {

"corr_npoints": 50,

"corr_precision":1.48e-6,

"cosmo_npoints": 50,

"cosmo_precision": 1.48e-8,

"dNdz_precision": 1.48e-8,

"halo_npoints": 50,

"halo_precision": 1.48e-5, ### This value is mostly due to integrations over

### the HOD. If you are intrested in dark matter

### only, this precision can be increased with

### no major hit to speed.

"halo_limit" : 100,

"kernel_npoints": 50,

"kernel_precision": 1.48e-6,

"kernel_limit": 100, ### If the variable force_quad is set in the Kernel

### class this value sets the limit for the quad

### integration

"kernel_bessel_limit": 8, ### Defines how many zeros before cutting off

### the Bessel function in kernel.py

"mass_npoints": 50,

"mass_precision": 1.48e-8,

"window_npoints": 50,

"window_precision": 1.48e-6,

"global_precision": 1.48e-32, ### Since the code has large range of values

### from say 1e-10 to 1e10 don't want to use

### absolute tolerances, instead using

### relative tolerances to define convergence

### of our integrands

"divmax":20

}

p_dict = {

}

### Fix cosmology used in the module in case the user changes the default

c_dict = {

}

c_dict_2 = {

}

h_dict = {

}

h_dict_2 = {

}

hod_dict = {

}

hod_dict_2 = {

}

degToRad = numpy.pi/180.0

### All precision values below come from running python2.7.2 in MacOS 10.7.4

### on a 1.7 GHz Intel Core i5

class CosmologyTestSingleEpoch(unittest.TestCase):

lin_power[idx], p_dict["cosmo"])

class CosmologyTestMultiEpoch(unittest.TestCase):

sigma_8_list[idx], p_dict["cosmo"])

class MassFunctionTest(unittest.TestCase):

f_mass_list[idx], p_dict["cosmo"])

class HODTest(unittest.TestCase):

self.nth_moment_list[idx])

class HaloTest(unittest.TestCase):

power_gg_list[idx], p_dict["halo"])

### Commented out currently as it is a future feature not yet mature.

# class HaloTriSpectrumTest(unittest.TestCase):

#

# def setUp(self):

# cosmo = cosmology.SingleEpoch(0.0, cosmo_dict=c_dict)

# mass = mass_function.MassFunctionSecondOrder(cosmo_single_epoch=cosmo,

# halo_dict=h_dict)

# pert = perturbation_spectra.PerturbationTheory(

# cosmo_single_epoch=cosmo)

# self.h = halo_trispectrum.HaloTrispectrum(

# redshift=0.0, single_epoch_cosmo=cosmo,

# mass_func_second=mass, perturbation=pert, halo_dict=h_dict)

# self.k_array = numpy.logspace(-3, 2, 4)

#

# def test_trispectrum(self):

# for idx, k in enumerate(self.k_array):

# self.assertGreater(

# self.h.trispectrum_parallelogram(k, k, 0.0), 0.0)

class dNdzTest(unittest.TestCase):

self.source_dist_list[idx])

class WindowFunctionTest(unittest.TestCase):

0.0), source_window_list[idx], p_dict["window"])

class KenrelTest(unittest.TestCase):

k_list[idx], p_dict["kernel"])

class CorrelationTest(unittest.TestCase):

corr_list[idx], p_dict["corr"])

if __name__ == "__main__":

print "*******************************"

print "* *"

print "* CHOMP Unit Test *"

print "* *"

print "*******************************"

unittest.main()