Reformatting

Author: mchan5

tests/integration/mock_drones/fast_drone.py

@@ -4,20 +4,18 @@
 """
 
 import time
-import sys
 from pymavlink import mavutil
 
 
-def main():
-    # Connect to the waiting telemetry script
-    # Note: Your test_telemetry.py uses tcp:localhost:12345
-    # The drone must LISTEN on this port or CONNECT to it depending on your setup.
-    # Based on standard MAVLink tests, usually the drone opens the server.
-
-    # We will try to bind to the port defined in your test_telemetry.py
+def main() -> None:  # <--- Added return type annotation here
+    """
+    Main execution loop.
+    Connects to the MAVLink receiver and sends high-frequency mock data.
+    """
     connection_string = "tcpin:localhost:12345"
     print(f"FAST DRONE: Waiting for connection on {connection_string}...")
 
+    # Initialize connection
     master = mavutil.mavlink_connection(connection_string)
     master.wait_heartbeat()
     print("FAST DRONE: Connected! BLASTING DATA AT 100Hz.")
@@ -29,12 +27,18 @@ def main():
 
         # 1. Send Attitude
         master.mav.attitude_send(
-            current_time_ms, 0.1, 0.2, 0.3, 0.01, 0.02, 0.03  # Roll, Pitch, Yaw  # Speeds
+            current_time_ms,
+            0.1,
+            0.2,
+            0.3,  # Roll, Pitch, Yaw (radians)
+            0.01,
+            0.02,
+            0.03,  # Roll, Pitch, Yaw speeds (rad/s)
         )
 
         # 2. Send Position
         master.mav.local_position_ned_send(
-            current_time_ms, 10.5, 20.2, -5.0, 0.5, 0.1, 0.0  # x, y, z  # vx, vy, vz
+            current_time_ms, 10.5, 20.2, -5.0, 0.5, 0.1, 0.0  # x, y, z (m)  # vx, vy, vz (m/s)
         )
 
         # SLEEP FOR 10ms = 100Hz