diff --git a/diagnostics/FullDiagnostics.py b/diagnostics/FullDiagnostics.py index 2a55560..12d5adb 100644 --- a/diagnostics/FullDiagnostics.py +++ b/diagnostics/FullDiagnostics.py @@ -7,9 +7,11 @@ HUB = PrimeHub() from BatteryDiagnostics import BatteryDiagnostics from MotorDiagnostics import MotorDiagnostics from ColorSensorDiagnostics import ColorSensorDiagnostics +from DriveBaseDiagnostics import DriveBaseDiagnostics battery = BatteryDiagnostics(HUB) motor = MotorDiagnostics(HUB, Motor) colorsensor = ColorSensorDiagnostics(HUB, ColorSensor) +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") @@ -56,5 +58,9 @@ while True: print("---------------------COLOR SENSOR DIAGNOSTICS---------------------") colorsensor.printAll() print("Color sensor diagnostics completed.") + elif choice == "db": + print("----------------------DRIVE BASE DIAGNOSTICS----------------------") + drivebase.printAll() + print("Drivebase diagnostics completed.") else: print("Invalid choice. Please enter 'b', 'm', or 'q'.") \ No newline at end of file