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()
