diagnostics/FullDiagnostics.py

@@ -7,13 +7,9 @@ HUB = PrimeHub()
 from BatteryDiagnostics import BatteryDiagnostics
 from MotorDiagnostics import MotorDiagnostics
 from ColorSensorDiagnostics import ColorSensorDiagnostics
-from DriveBaseDiagnostics import DriveBaseDiagnostics
-from HubDiagnostics import HubDiagnostics
 battery = BatteryDiagnostics(HUB)
 motor = MotorDiagnostics(HUB, Motor)
 colorsensor = ColorSensorDiagnostics(HUB, ColorSensor)
-hubdiags = HubDiagnostics(HUB)
-drivebase = DriveBaseDiagnostics(HUB, Motor, DriveBase)
 CLEARCONFIRM = input("Clear the console before proceeding? Y/N (default: yes): ")
 if(CLEARCONFIRM == "Y" or CLEARCONFIRM == "y" or CLEARCONFIRM == "yes" or CLEARCONFIRM == ""):
     print("Clearing console... \n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n")
@@ -27,9 +23,8 @@ print("""
  █████           █████    █████  ▒▒█████ █████   █████ █████     █████ █████ ▒▒█████████ ▒▒█████████ 
 ▒▒▒▒▒           ▒▒▒▒▒    ▒▒▒▒▒    ▒▒▒▒▒ ▒▒▒▒▒   ▒▒▒▒▒ ▒▒▒▒▒     ▒▒▒▒▒ ▒▒▒▒▒   ▒▒▒▒▒▒▒▒▒   ▒▒▒▒▒▒▒▒▒  
                                                                                                      
-The free and open source diagnostics tool for the LEGO® Education SPIKE™ Prime robots, designed for FIRST Lego League.
+The free and open source diagnostics tool for LEGO® Education SPIKE™ Prime robots, designed for FIRST Lego League.
 Developed by Team 65266, Lego Dynamics.                                                
-Please set your window size to 90% on small screens for best results with the ASCII art.               
           """
           )
 while True:
@@ -38,8 +33,6 @@ while True:
     print("Enter 'b' for battery diagnostics")
     print("Enter 'm' for motor diagnostics")
     print("Enter 'cs' for color sensor diagnostics")
-    print("Enter 'h' for hub diagnostics")
-    print("Enter 'db' for drive base diagnostics")
     print("Enter 'q' to quit")
 
     choice = input("Your choice: ").strip().lower()
@@ -55,20 +48,13 @@ while True:
         print("------------------------MOTOR DIAGNOSTICS------------------------")
         motor.fullTest()
         print("Motor diagnostics completed.")
-    elif choice == "h":
+
-        print("-------------------------HUB DIAGNOSTICS-------------------------")
-        hubdiags.printAll()
-        print("Hub diagnostics completed.")
     elif choice == "q":
-        print("Diagnostics completed successfully. Exiting with code 0. Good luck in the robot game!")
+        print("Diagnostics completed successfully. Exiting program.")
         break
-    elif choice == "db":
-        print("----------------------DRIVEBASE DIAGNOSTICS----------------------")
-        drivebase.printAll()
     elif choice == "cs":
         print("---------------------COLOR SENSOR DIAGNOSTICS---------------------")
         colorsensor.printAll()
         print("Color sensor diagnostics completed.")
-        
     else:
         print("Invalid choice. Please enter 'b', 'm', or 'q'.")