add on enable and on disable to test

Author: LucienMorey

magicbot/magiccomponent.py

@@ -49,7 +49,7 @@ def setup(self) -> None:
 
     def on_enable(self) -> None:
         """
-        Called when the robot enters autonomous or teleoperated mode. This
+        Called when the robot enters autonomous, teleoperated or test mode mode. This
         function should initialize your component to a "safe" state so
         that unexpected things don't happen when enabling the robot.
 
@@ -60,7 +60,7 @@ def on_enable(self) -> None:
 
     def on_disable(self) -> None:
         """
-        Called when the robot leaves autonomous or teleoperated
+        Called when the robot leaves autonomous, teleoperated or test mode
         """
 
     def execute(self) -> None:

magicbot/magicrobot.py

@@ -526,6 +526,9 @@ def _test(self) -> None:
         wpilib.LiveWindow.setEnabled(True)
         # Shuffleboard.enableActuatorWidgets()
 
+        # initialize things
+        self._on_mode_enable_components()
+
         try:
             self.testInit()
         except:
@@ -559,6 +562,8 @@ def _test(self) -> None:
         wpilib.LiveWindow.setEnabled(False)
         # Shuffleboard.disableActuatorWidgets()
 
+        self._on_mode_disable_components()
+
     def _on_mode_enable_components(self) -> None:
         # initialize things
         for _, component in self._components: