Spaces:
Running
Running
File size: 5,065 Bytes
bd61f34 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 |
#!/usr/bin/env python3
"""
Simplified test script for mission planning and execution.
This script tests the mission planning and execution functionality directly,
without using the DroneAssistant or smolagents.
"""
import time
import math
import sys
import json
from drone.drone_chat import generate_mission_plan
from drone.drone_control import connect_drone, disconnect_drone, takeoff, land, return_home
from drone import compatibility_fix # Import for Python 3.10+ compatibility
def test_mission_planning():
"""Test the mission planning functionality."""
print("\n----- TESTING MISSION PLANNING -----")
# Test different mission types
mission_types = ["survey", "inspection", "delivery", "custom"]
durations = [10, 15, 20, 5]
for mission_type, duration in zip(mission_types, durations):
print(f"\nGenerating {mission_type} mission plan for {duration} minutes...")
mission_plan = generate_mission_plan(mission_type=mission_type, duration_minutes=duration)
# Parse the mission plan
if isinstance(mission_plan, str):
try:
plan = eval(mission_plan) # Convert string representation to dict
print("Successfully parsed mission plan:")
print(f" Mission type: {plan.get('mission_type')}")
print(f" Duration: {plan.get('duration_minutes')} minutes")
print(f" Flight pattern: {plan.get('flight_pattern')}")
print(f" Recommended altitude: {plan.get('recommended_altitude')} meters")
print(f" Waypoint count: {len(plan.get('waypoints', []))}")
print(" Mission description:", plan.get('description', '')[:50] + "...")
except Exception as e:
print(f"Error parsing mission plan: {str(e)}")
print("Raw mission plan:", mission_plan)
else:
print("Mission plan is not a string:", type(mission_plan))
print(mission_plan)
print("\nMission planning test completed.")
return True
def test_mission_execution(mission_type="survey", duration=10):
"""Test the mission execution functionality using the simulator."""
print("\n----- TESTING MISSION EXECUTION -----")
# Generate a mission plan
print(f"Generating {mission_type} mission plan for {duration} minutes...")
mission_plan = generate_mission_plan(mission_type=mission_type, duration_minutes=duration)
# Parse the mission plan
try:
if isinstance(mission_plan, str):
plan = eval(mission_plan) # Convert string representation to dict
else:
plan = mission_plan
print("Mission plan details:")
print(f" Mission type: {plan.get('mission_type')}")
print(f" Duration: {plan.get('duration_minutes')} minutes")
print(f" Flight pattern: {plan.get('flight_pattern')}")
print(f" Recommended altitude: {plan.get('recommended_altitude')} meters")
print(f" Waypoint count: {len(plan.get('waypoints', []))}")
# Connect to the simulator
print("\nConnecting to simulator...")
success = connect_drone('udp:127.0.0.1:14550')
if not success:
print("Failed to connect to the simulator. Make sure it's running.")
return False
print("Connected to simulator!")
# Execute the mission
# For the test, we'll just print the waypoints instead of actually flying
print("\nSimulating mission execution...")
print(f"Taking off to {plan.get('recommended_altitude')} meters...")
time.sleep(2)
print("Flying mission waypoints:")
waypoints = plan.get('waypoints', [])
for i, waypoint in enumerate(waypoints):
print(f" Waypoint {i+1}: {waypoint}")
time.sleep(1)
print("Returning to home...")
time.sleep(2)
print("Landing...")
time.sleep(2)
print("Mission completed successfully!")
disconnect_drone()
return True
except Exception as e:
print(f"Error executing mission: {str(e)}")
return False
def main():
"""Run all tests."""
print("Testing mission planning and execution...")
# Test mission planning
planning_success = test_mission_planning()
# Test mission execution
execution_success = test_mission_execution("survey", 10)
# Print test summary
print("\n----- TEST SUMMARY -----")
print(f"Mission Planning Test: {'PASS' if planning_success else 'FAIL'}")
print(f"Mission Execution Test: {'PASS' if execution_success else 'FAIL'}")
if planning_success and execution_success:
print("\nAll tests passed! The mission planning and execution are working properly.")
return 0
else:
print("\nSome tests failed. Check the output above for details.")
return 1
if __name__ == "__main__":
sys.exit(main()) |