def test_modules(self):
     chain = LikelihoodComputationChain()
     
     assert len(chain.getCoreModules())==0
     assert len(chain.getLikelihoodModules())==0
     
     coreModule = DummyModule()
     likeModule = DummyModule()
     chain.addCoreModule(coreModule)
     chain.addLikelihoodModule(likeModule)
     assert len(chain.getCoreModules())==1
     assert len(chain.getLikelihoodModules())==1
     
     chain.setup()
     assert coreModule.init
     assert likeModule.init
     
     like, data = chain([0])
     
     assert coreModule.called
     assert likeModule.compLike
     
     assert like == DummyModule.like
     assert len(data) == 1
     assert data["data"] == DummyModule.data
Beispiel #2
0
    def test_modules(self):
        chain = LikelihoodComputationChain()

        assert len(chain.getCoreModules()) == 0
        assert len(chain.getLikelihoodModules()) == 0

        coreModule = DummyModule()
        likeModule = DummyModule()
        chain.addCoreModule(coreModule)
        chain.addLikelihoodModule(likeModule)
        assert len(chain.getCoreModules()) == 1
        assert len(chain.getLikelihoodModules()) == 1

        chain.setup()
        assert coreModule.init
        assert likeModule.init

        like, data = chain([0])

        assert coreModule.called
        assert likeModule.compLike

        assert like == DummyModule.like
        assert len(data) == 1
        assert data["data"] == DummyModule.data