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)
#!/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)