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