Skip to content

Commit fdcf570

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

File tree

1 file changed

+15
-2
lines changed

1 file changed

+15
-2
lines changed

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

+15-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,9 +108,15 @@ 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)
119+
intertialAttitudeFilter.imuSensorDataInMsg.subscribeTo(miru_low_pass_filter_converter.imuSensorOutMsg)
113120
unit_test_sim.AddModelToTask(unit_task_name, intertialAttitudeFilter)
114121

115122
# Add test module to runtime call list
@@ -173,7 +180,7 @@ def test_propagation_kf(show_plots):
173180

174181
accel_data = messaging.AccDataMsgPayload()
175182
accel_measurement = messaging.AccDataMsg().write(accel_data)
176-
intertialAttitudeFilter.accelDataMsg.subscribeTo(accel_measurement)
183+
miru_low_pass_filter_converter.imuAccelDataInMsg.subscribeTo(accel_measurement)
177184

178185
attitude_data_log = intertialAttitudeFilter.inertialFilterOutputMsg.recorder()
179186
unit_test_sim.AddModelToTask(unit_task_name, attitude_data_log)
@@ -226,7 +233,13 @@ def test_measurements_kf(show_plots, initial_error, method):
226233
test_process = unit_test_sim.CreateNewProcess(unit_process_name)
227234
test_process.addTask(unit_test_sim.CreateNewTask(unit_task_name, test_process_rate))
228235

236+
# Create miruLowPassFilterConverter module
237+
miru_low_pass_filter_converter = miruLowPassFilterConverter.MiruLowPassFilterConverter()
238+
miru_low_pass_filter_converter.setLowPassFilter(0.5, 15/(2*np.pi))
239+
unit_test_sim.AddModelToTask(unit_task_name, miru_low_pass_filter_converter)
240+
229241
intertialAttitudeFilter = inertialAttitudeUkf.InertialAttitudeUkf(method)
242+
intertialAttitudeFilter.imuSensorDataInMsg.subscribeTo(miru_low_pass_filter_converter.imuSensorOutMsg)
230243
unit_test_sim.AddModelToTask(unit_task_name, intertialAttitudeFilter)
231244

232245
# Add test module to runtime call list
@@ -294,7 +307,7 @@ def test_measurements_kf(show_plots, initial_error, method):
294307

295308
accel_data = messaging.AccDataMsgPayload()
296309
accel_measurement = messaging.AccDataMsg().write(accel_data)
297-
intertialAttitudeFilter.accelDataMsg.subscribeTo(accel_measurement)
310+
miru_low_pass_filter_converter.imuAccelDataInMsg.subscribeTo(accel_measurement)
298311

299312
filter_data_log = intertialAttitudeFilter.inertialFilterOutputMsg.recorder()
300313
unit_test_sim.AddModelToTask(unit_task_name, filter_data_log)

0 commit comments

Comments
 (0)