Skip to content

Commit

Permalink
Tools: Autotest: create gps-for-yaw test for Sub
Browse files Browse the repository at this point in the history
  • Loading branch information
Williangalvani committed Nov 28, 2023
1 parent 3c0acd0 commit d437f6c
Show file tree
Hide file tree
Showing 2 changed files with 46 additions and 1 deletion.
40 changes: 39 additions & 1 deletion Tools/autotest/ardusub.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,11 +9,12 @@
import sys
import time

from pymavlink import mavutil
from pymavlink import mavutil, mavextra

import vehicle_test_suite
from vehicle_test_suite import NotAchievedException
from vehicle_test_suite import AutoTestTimeoutException
from math import degrees

if sys.version_info[0] < 3:
ConnectionResetError = AutoTestTimeoutException
Expand Down Expand Up @@ -474,6 +475,42 @@ def MAV_CMD_DO_CHANGE_SPEED(self):
self.wait_groundspeed(speed-0.2, speed+0.2, minimum_duration=2, timeout=60)
self.disarm_vehicle()

def GPSForYaw(self):
'''Moving baseline GPS yaw'''
# store previous heading
previous_heading = self.assert_receive_message("ATTITUDE").yaw

# load parameters with gps for yaw and 10 degrees offset
self.context_push()
self.load_default_params_file("sub-gps-for-yaw-nmea.parm")
self.reboot_sitl()
ex = None
try:
self.wait_gps_fix_type_gte(3, message_type="GPS_RAW_INT", verbose=True)
gps_m = self.assert_receive_message("GPS_RAW_INT")
attitude_m = self.assert_receive_message("ATTITUDE")
self.progress(self.dump_message_verbose(gps_m))
self.wait_heading(195, minimum_duration=2)
want = mavextra.wrap_360(degrees(previous_heading) + 10) * 100
attitude_yaw_deg = mavextra.wrap_360(degrees(attitude_m.yaw)) * 100
# ensure new reading propagated to GPS_RAW_INT
if abs(gps_m.yaw - want) > 500:
raise NotAchievedException("Expected to get yaw populated on GPS_RAW_INT (want %f got %f)" % (want, gps_m.yaw))
# ensure new reading propagated to ATTITUDE
if abs(attitude_yaw_deg - want) > 500:
raise NotAchievedException(
"Expected to get yaw consumed and at ATTITUDE (want %f got %f)" % (want, attitude_yaw_deg)
)
self.wait_ready_to_arm()
except Exception as e:
self.print_exception_caught(e)
ex = e

self.context_pop()
self.reboot_sitl()
if ex is not None:
raise ex

def _MAV_CMD_CONDITION_YAW(self, run_cmd):
self.arm_vehicle()
self.change_mode('GUIDED')
Expand Down Expand Up @@ -536,6 +573,7 @@ def tests(self):
self.MAV_CMD_MISSION_START,
self.MAV_CMD_DO_CHANGE_SPEED,
self.MAV_CMD_CONDITION_YAW,
self.GPSForYaw,
])

return ret
7 changes: 7 additions & 0 deletions Tools/autotest/default_params/sub-gps-for-yaw-nmea.parm
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
# SITL GPS-for-yaw using a single simulated NMEA GPS
EK3_SRC1_YAW 3
GPS_TYPE 5
SIM_GPS_TYPE 5
SIM_GPS_HDG 1
GPS_AUTO_CONFIG 0
SIM_GPS_HDG_OFS 10.0

0 comments on commit d437f6c

Please sign in to comment.