Update diagnostics/HubDiagnostics.py

diagnostics/HubDiagnostics.py

@@ -4,12 +4,30 @@ from pybricks.parameters import Button, Color, Direction, Port, Side, Stop
 from pybricks.robotics import DriveBase
 from pybricks.tools import wait, StopWatch
 from pybricks import version
+from OtherFunctions import vprint
 import usys
-hub = PrimeHub()
 
 print("Pybricks version information:", version)
 print("MicroPython information:", usys.implementation)
 print("MicroPython version:", usys.version)
-while True:
+class HubDiagnostics:
-    print(str(hub.imu.heading()))
+    def __init__(self):
-    # Insert display test here, with input to either test hub gyro stats or display or info.
+        self.hub = PrimeHub()
+        self.port_map = {
+            "A": Port.A,
+            "B": Port.B,
+            "C": Port.C,
+            "D": Port.D,
+            "E": Port.E,
+            "F": Port.F,
+        }
+    def testLightSources(self, verbose):
+        v = verbose
+        hub.display.off()
+        for x in range(5):
+            for y in range(5):
+                vprint(f"Turning on pixel at position {x}, {y}...", v)
+                display.pixel(x, y, brightness=100)
+                wait(100)
+                vprint(f"Turning off pixel at position {x}, {y}...", v)
+                display.pixel(x, y, brightness=0)