Python HTMModuleFactory.run Examples

Example #1

File: test_htmModel.py Project: antsago/3rdYearProject

  def test_whenMakeAMoveSequenceIsInvoked(self):
    mockedModule = HTMModuleFactory().createModule(["A"])
    mockedModule.run = MagicMock()
    model = HTMModel(mockedModule)
    
    model.makeMove("A", 0, "A")

    mockedModule.run.assert_called_once()

Example #2

File: test_htmModel.py Project: antsago/3rdYearProject

  def test_whenPredictingRewardsModuleIsSaveAndLoadedForEachPossibleState(self):
    mockedModule = HTMModuleFactory().createModule(["A", "B"])
    mockedModule.run = MagicMock()
    mockedModule.load = MagicMock()
    mockedModule.save = MagicMock()
    model = HTMModel(mockedModule)
    
    model.predictRewards("A", 0, ["A", "B"])

    mockedModule.run.assert_called_twice()
    mockedModule.save.assert_called_once()
    mockedModule.load.assert_called_twice()