def test_step_without_update(self): controller = DummyController() view = View(controller) controller.should_update = MagicMock(return_value=False) controller.update = MagicMock(side_effect=Exception) controller.step = MagicMock(return_value='action') self.assertEqual(view.step('obs', 'reward', 'done', 'info'), 'action') controller.step.assert_called_once_with('obs', 'reward', 'done', 'info')
def test_step_without_save(self): controller = DummyController() view = View(controller) controller.should_save = MagicMock(return_value=True) controller.save = MagicMock(unsafe=True) controller.step = MagicMock(return_value='action') self.assertEqual(view.step('obs', 'reward', 'done', 'info'), 'action') controller.step.assert_called_once_with('obs', 'reward', 'done', 'info') controller.save.assert_called_once_with()