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