def test_whenProblemFinishedResetSequenceIsCalled(self): mockedModule = HTMModuleFactory().createModule(["A"]) mockedModule.resetSequenceStates = MagicMock() model = HTMModel(mockedModule) model.problemFinished() mockedModule.resetSequenceStates.resetSequenceStates.assert_called_once()