def createChainContext(self, p):
"""
Returns a new instance of a chain context
"""
try:
p = Params(*zip(self.params.keys, p))
except Exception:
# no params or params has no keys
pass
return ChainContext(self, p)
wmap9Likelihood = wmap9.WmapLikelihoodModule()
#setting up wrappers
cambCore.setup()
pyCambCore.setup()
wmap3Likelihood.setup()
wmap5Likelihood.setup()
wmap7Likelihood.setup()
wmap9Likelihood.setup()
#camb wrapper
print("setup done. calling CAMB power spectrum computation")
paramValues = [70, 0.0226, 0.122, 2.1E-009, 0.96, 0.09]
ctx = ChainContext(None, paramValues)
cambCore(ctx)
print("cls computation done. calling WMAP likelihood")
print("Likelihood WMAP 3: %s" % wmap3Likelihood.computeLikelihood(ctx))
print("Likelihood WMAP 5: %s" % wmap5Likelihood.computeLikelihood(ctx))
print("Likelihood WMAP 7: %s" % wmap7Likelihood.computeLikelihood(ctx))
print("Likelihood WMAP 9: %s" % wmap9Likelihood.computeLikelihood(ctx))
#pycamb
ctx = ChainContext(None, paramValues)
pyCambCore(ctx)
print("cls computation done. calling WMAP likelihood")
print("Likelihood WMAP 3: %s" % wmap3Likelihood.computeLikelihood(ctx))
print("Likelihood WMAP 5: %s" % wmap5Likelihood.computeLikelihood(ctx))
#!/usr/bin/env python
import pylab
from cosmoHammer.ChainContext import ChainContext
import pycambWrapper
paramValues = [70, 0.0426, 0.122, 2.1E-009, 0.96, 0.09]
ctx = ChainContext(None, paramValues)
pyCambCore = pycambWrapper.PyCambCoreModule()
pyCambCore(ctx)
cl_tt = ctx.get(pycambWrapper.CL_TT_KEY)
cl_te = ctx.get(pycambWrapper.CL_TE_KEY)
cl_ee = ctx.get(pycambWrapper.CL_EE_KEY)
cl_bb = ctx.get(pycambWrapper.CL_BB_KEY)
pylab.subplot(1, 1, 1)
pylab.title("Power spectrum")
ell = pylab.arange(1, len(cl_tt) + 1)
pylab.semilogx(ell, cl_tt)
pylab.xlabel("$\ell$", fontsize=20)
pylab.ylabel("$\ell (\ell+1) C_\ell / 2\pi \quad [\mu K^2]$", fontsize=20)
pylab.xlim(1, 2000)
pylab.show()
def setup(self):
self.ctx=ChainContext(self, self.params)
#!/usr/bin/env python from cambWrapper import CambCoreModule, CL_TT_KEY, CL_TE_KEY, CL_EE_KEY, CL_BB_KEY from cosmoHammer.ChainContext import ChainContext lmax = 2000 camb = CambCoreModule(lmax = lmax) camb.setup() params = [70, 0.0426, 0.122, 2.1E-009, 0.96, 0.09] ctx = ChainContext(None, params) camb(ctx) cl_tt =ctx.get(CL_TT_KEY) assert cl_tt.shape == (lmax-1,) assert ctx.contains(CL_TE_KEY) assert ctx.contains(CL_EE_KEY) assert ctx.contains(CL_BB_KEY)
#!/usr/bin/env python
import pylab
from cosmoHammer.ChainContext import ChainContext
import pycambWrapper
paramValues = [70, 0.0426, 0.122, 2.1E-009, 0.96, 0.09]
ctx = ChainContext(None, paramValues)
pyCambCore = pycambWrapper.PyCambCoreModule()
pyCambCore(ctx)
cl_tt =ctx.get(pycambWrapper.CL_TT_KEY)
cl_te =ctx.get(pycambWrapper.CL_TE_KEY)
cl_ee =ctx.get(pycambWrapper.CL_EE_KEY)
cl_bb =ctx.get(pycambWrapper.CL_BB_KEY)
pylab.subplot(1,1,1)
pylab.title("Power spectrum")
ell = pylab.arange(1,len(cl_tt)+1)
pylab.semilogx(ell,cl_tt)
pylab.xlabel("$\ell$", fontsize=20)
pylab.ylabel("$\ell (\ell+1) C_\ell / 2\pi \quad [\mu K^2]$", fontsize=20)
pylab.xlim(1,2000)
pylab.show()
def createChainContext(self, p):
"""
Returns a new instance of a chain context
"""
return ChainContext(self, p)
camb = PyCambCoreModule(lmax=2700)
camb.setup()
# Path to clik files
path = ''
clik_files = [
join(path, 'commander_v4.1_lm49.clik'),
join(path, 'CAMspec_v6.2TN_2013_02_26_dist.clik'),
join(path, 'lowlike_v222.clik')
]
# clik_files = join(path,'commander_v4.1_lm49.clik')
params = array([
67.04, 0.022032, 0.12038,
exp(3.0980) * 1e-10, 0.9619, 0.0925, 152, 63.3, 117.0, 0.0, 27.2, 6.80,
0.916, 0.406, 0.601, 1.0, 1.0, 0.03, 0.9, 0.0
])
nuisance_indices = [[], range(6, 6 + 14), []]
# nuisance_indices = [[]]
like = PlanckLikelihoodModule(clik_files, nuisance_indices)
like.setup()
ctx = ChainContext(None, params)
camb(ctx)
print like.computeLikelihood(ctx)
class TestCosmoHammerSampler(object):
ctx = None
params = np.array([[1, 2, 3], [4, 5, 6]])
def setup(self):
self.ctx = ChainContext(self, self.params)
def test_no_data(self):
assert self.ctx.getParent() == self
assert self.ctx.getParams() is self.params
assert self.ctx.contains("key") == False
assert self.ctx.getData() is not None
assert len(list(self.ctx.getData().items())) == 0
def test_add_data(self):
assert self.ctx.contains("key") == False
assert self.ctx.getData() is not None
self.ctx.add("key", "value")
assert self.ctx.contains("key") == True
assert self.ctx.getData() is not None
assert self.ctx.get("key") == "value"
def test_remove_data(self):
assert self.ctx.contains("key2") == False
assert self.ctx.getData() is not None
self.ctx.add("key2", "value")
assert self.ctx.contains("key2") == True
assert self.ctx.getData() is not None
assert self.ctx.get("key2") == "value"
self.ctx.remove("key2")
assert self.ctx.contains("key2") == False
def test_get_default(self):
assert self.ctx.contains("aaa") == False
assert self.ctx.get("aaa", "default") == "default"
def test_additional_data(self):
assert self.ctx.getData() is not None
assert len(list(self.ctx.getData().items())) == 0
self.ctx.getData()["moreData"] = "data"
assert len(list(self.ctx.getData().items())) == 1
def setup(self):
self.ctx=ChainContext(self, self.params)
class TestCosmoHammerSampler(object):
ctx = None
params = np.array([[1,2,3],[4,5,6]])
def setup(self):
self.ctx=ChainContext(self, self.params)
def test_no_data(self):
assert self.ctx.getParent() == self
assert self.ctx.getParams() is self.params
assert self.ctx.contains("key") == False
assert self.ctx.getData() is not None
assert len(list(self.ctx.getData().items())) == 0
def test_add_data(self):
assert self.ctx.contains("key") == False
assert self.ctx.getData() is not None
self.ctx.add("key", "value")
assert self.ctx.contains("key") == True
assert self.ctx.getData() is not None
assert self.ctx.get("key") == "value"
def test_remove_data(self):
assert self.ctx.contains("key2") == False
assert self.ctx.getData() is not None
self.ctx.add("key2", "value")
assert self.ctx.contains("key2") == True
assert self.ctx.getData() is not None
assert self.ctx.get("key2") == "value"
self.ctx.remove("key2")
assert self.ctx.contains("key2") == False
def test_get_default(self):
assert self.ctx.contains("aaa") == False
assert self.ctx.get("aaa", "default") == "default"
def test_additional_data(self):
assert self.ctx.getData() is not None
assert len(list(self.ctx.getData().items())) == 0
self.ctx.getData()["moreData"] = "data"
assert len(list(self.ctx.getData().items())) == 1
#!/usr/bin/env python from cambWrapper import CambCoreModule, CL_TT_KEY, CL_TE_KEY, CL_EE_KEY, CL_BB_KEY from cosmoHammer.ChainContext import ChainContext lmax = 2000 camb = CambCoreModule(lmax=lmax) camb.setup() params = [70, 0.0426, 0.122, 2.1E-009, 0.96, 0.09] ctx = ChainContext(None, params) camb(ctx) cl_tt = ctx.get(CL_TT_KEY) assert cl_tt.shape == (lmax - 1, ) assert ctx.contains(CL_TE_KEY) assert ctx.contains(CL_EE_KEY) assert ctx.contains(CL_BB_KEY)
