def test_toggle():
shooter = Shooter()
shooter.shooter_motor = MagicMock()
shooter.state = States.off
shooter.changed_state = False
shooter.toggle()
shooter.execute()
shooter.execute()
assert shooter._speed != 0.0
shooter.state = States.shooting
shooter.changed_state = False
shooter.execute()
shooter.toggle()
shooter.execute()
shooter.execute()
assert shooter._speed == 0.0
assert shooter.shooter_motor.set.call_count == 2
def test_backdrive():
shooter = Shooter()
shooter.shooter_motor = MagicMock()
shooter.state = States.backdriving
shooter.execute()
assert shooter._speed >= 0.0
