fix whileHeld instant command

Author: TheTripleV

commands2/coroutinecommand.py

@@ -28,6 +28,7 @@ def ensure_generator_function(func: Coroutineable) -> Callable[..., Coroutine]:
     @wraps(func)
     def wrapper(*args, **kwargs):
         func(*args, **kwargs)
+        return
         yield
 
     return wrapper
@@ -81,6 +82,7 @@ def initialize(self) -> None:
         self.is_finished = False
 
     def execute(self):
+        print("r")
         try:
             if not self.is_finished:
                 if not self.coroutine:

tests/test_button.py

@@ -161,7 +161,7 @@ def test_cancel_when_pressed(scheduler: commands2.CommandScheduler):
     assert cmd1.canceled == 1
     assert not scheduler.isScheduled(cmd1)
 
-@pytest.mark.xfail(strict=False)
+
 def test_function_bindings(scheduler: commands2.CommandScheduler):
 
     buttonWhenPressed = MyButton()
@@ -174,9 +174,9 @@ def test_function_bindings(scheduler: commands2.CommandScheduler):
 
     counter = Counter()
 
-    buttonWhenPressed.whenPressed(lambda: (print("wp"), counter.increment(), None)[2])
-    buttonWhileHeld.whileHeld(lambda: (print("wh"), counter.increment(), None)[2])
-    buttonWhenReleased.whenReleased(lambda: (print("wr"), counter.increment(), None)[2])
+    buttonWhenPressed.whenPressed(counter.increment)
+    buttonWhileHeld.whileHeld(counter.increment)
+    buttonWhenReleased.whenReleased(counter.increment)
 
     scheduler.run()
     buttonWhenPressed.setPressed(True)