Skip to content

Commit 5be4d1f

Browse files
committed
Update inertialAttitudeUkf unit test
1 parent 78367c4 commit 5be4d1f

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

src/fswAlgorithms/attDetermination/inertialAttitudeUkf/tests/test_inertialAttitudeUkf.py

+17-2
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import pytest
2121
from Basilisk.architecture import messaging
2222
from Basilisk.fswAlgorithms import inertialAttitudeUkf
23+
from Basilisk.fswAlgorithms import miruLowPassFilterConverter
2324
from Basilisk.utilities import SimulationBaseClass, macros
2425
from Basilisk.utilities import RigidBodyKinematics as rbk
2526

@@ -107,6 +108,11 @@ def test_propagation_kf(show_plots):
107108
test_process = unit_test_sim.CreateNewProcess(unit_process_name)
108109
test_process.addTask(unit_test_sim.CreateNewTask(unit_task_name, test_process_rate))
109110

111+
# Create miruLowPassFilterConverter module
112+
miru_low_pass_filter_converter = miruLowPassFilterConverter.MiruLowPassFilterConverter()
113+
miru_low_pass_filter_converter.setLowPassFilter(0.5, 15/(2*np.pi))
114+
unit_test_sim.AddModelToTask(unit_task_name, miru_low_pass_filter_converter)
115+
110116
# Construct algorithm and associated C++ container
111117
allMeasurements = inertialAttitudeUkf.AttitudeFilterMethod_AllMeasurements
112118
intertialAttitudeFilter = inertialAttitudeUkf.InertialAttitudeUkf(allMeasurements)
@@ -173,7 +179,7 @@ def test_propagation_kf(show_plots):
173179

174180
accel_data = messaging.AccDataMsgPayload()
175181
accel_measurement = messaging.AccDataMsg().write(accel_data)
176-
intertialAttitudeFilter.accelDataMsg.subscribeTo(accel_measurement)
182+
miru_low_pass_filter_converter.imuAccelDataInMsg.subscribeTo(accel_measurement)
177183

178184
attitude_data_log = intertialAttitudeFilter.inertialFilterOutputMsg.recorder()
179185
unit_test_sim.AddModelToTask(unit_task_name, attitude_data_log)
@@ -226,6 +232,11 @@ def test_measurements_kf(show_plots, initial_error, method):
226232
test_process = unit_test_sim.CreateNewProcess(unit_process_name)
227233
test_process.addTask(unit_test_sim.CreateNewTask(unit_task_name, test_process_rate))
228234

235+
# Create miruLowPassFilterConverter module
236+
miru_low_pass_filter_converter = miruLowPassFilterConverter.MiruLowPassFilterConverter()
237+
miru_low_pass_filter_converter.setLowPassFilter(0.5, 15/(2*np.pi))
238+
unit_test_sim.AddModelToTask(unit_task_name, miru_low_pass_filter_converter)
239+
229240
intertialAttitudeFilter = inertialAttitudeUkf.InertialAttitudeUkf(method)
230241
unit_test_sim.AddModelToTask(unit_task_name, intertialAttitudeFilter)
231242

@@ -292,9 +303,13 @@ def test_measurements_kf(show_plots, initial_error, method):
292303
star_tracker2.measurementNoise = [[1e-4, 0, 0], [0,1e-4,0], [0,0,1e-4]]
293304
intertialAttitudeFilter.addStarTrackerInput(star_tracker2)
294305

306+
imu_sensor_msg = messaging.IMUSensorMsgPayload()
307+
imu_sensor_message = messaging.IMUSensorMsg().write(imu_sensor_msg)
308+
intertialAttitudeFilter.imuSensorDataInMsg.subscribeTo(imu_sensor_message)
309+
295310
accel_data = messaging.AccDataMsgPayload()
296311
accel_measurement = messaging.AccDataMsg().write(accel_data)
297-
intertialAttitudeFilter.accelDataMsg.subscribeTo(accel_measurement)
312+
miru_low_pass_filter_converter.imuAccelDataInMsg.subscribeTo(accel_measurement)
298313

299314
filter_data_log = intertialAttitudeFilter.inertialFilterOutputMsg.recorder()
300315
unit_test_sim.AddModelToTask(unit_task_name, filter_data_log)

0 commit comments

Comments
 (0)