def test_execute(): defeater = Defeater() defeater.defeater_motor = MagicMock() defeater.defeater_motor.set = MagicMock() defeater.defeater_motor.getOutputCurrent = MagicMock(return_value=9.0) defeater.execute() assert defeater.defeater_motor.set.not_called defeater.defeater_motor.getOutputCurrent = MagicMock(return_value=11.0) defeater.execute() defeater.defeater_motor.set.assert_called_with(0.0)