def test_forward_stop(self, io): controller = MotorController() controller.progress_keys.add(UP_KEY) controller.forward() calls = [call(4, False), call(17, False)] io.output.assert_has_calls(calls) self.assertEqual(controller.state, 'stopped')
def test_forward(self, io): controller = MotorController() controller.forward() calls = [call(4, True), call(17, False)] io.output.assert_has_calls(calls) self.assertEqual(controller.state, 'forward')