|
20 | 20 | import pytest
|
21 | 21 | from Basilisk.architecture import messaging
|
22 | 22 | from Basilisk.fswAlgorithms import inertialAttitudeUkf
|
| 23 | +from Basilisk.fswAlgorithms import miruLowPassFilterConverter |
23 | 24 | from Basilisk.utilities import SimulationBaseClass, macros
|
24 | 25 | from Basilisk.utilities import RigidBodyKinematics as rbk
|
25 | 26 |
|
@@ -107,9 +108,15 @@ def test_propagation_kf(show_plots):
|
107 | 108 | test_process = unit_test_sim.CreateNewProcess(unit_process_name)
|
108 | 109 | test_process.addTask(unit_test_sim.CreateNewTask(unit_task_name, test_process_rate))
|
109 | 110 |
|
| 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 | + |
110 | 116 | # Construct algorithm and associated C++ container
|
111 | 117 | allMeasurements = inertialAttitudeUkf.AttitudeFilterMethod_AllMeasurements
|
112 | 118 | intertialAttitudeFilter = inertialAttitudeUkf.InertialAttitudeUkf(allMeasurements)
|
| 119 | + intertialAttitudeFilter.imuSensorDataInMsg.subscribeTo(miru_low_pass_filter_converter.imuSensorOutMsg) |
113 | 120 | unit_test_sim.AddModelToTask(unit_task_name, intertialAttitudeFilter)
|
114 | 121 |
|
115 | 122 | # Add test module to runtime call list
|
@@ -173,7 +180,7 @@ def test_propagation_kf(show_plots):
|
173 | 180 |
|
174 | 181 | accel_data = messaging.AccDataMsgPayload()
|
175 | 182 | accel_measurement = messaging.AccDataMsg().write(accel_data)
|
176 |
| - intertialAttitudeFilter.accelDataMsg.subscribeTo(accel_measurement) |
| 183 | + miru_low_pass_filter_converter.imuAccelDataInMsg.subscribeTo(accel_measurement) |
177 | 184 |
|
178 | 185 | attitude_data_log = intertialAttitudeFilter.inertialFilterOutputMsg.recorder()
|
179 | 186 | unit_test_sim.AddModelToTask(unit_task_name, attitude_data_log)
|
@@ -226,7 +233,13 @@ def test_measurements_kf(show_plots, initial_error, method):
|
226 | 233 | test_process = unit_test_sim.CreateNewProcess(unit_process_name)
|
227 | 234 | test_process.addTask(unit_test_sim.CreateNewTask(unit_task_name, test_process_rate))
|
228 | 235 |
|
| 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 | + |
229 | 241 | intertialAttitudeFilter = inertialAttitudeUkf.InertialAttitudeUkf(method)
|
| 242 | + intertialAttitudeFilter.imuSensorDataInMsg.subscribeTo(miru_low_pass_filter_converter.imuSensorOutMsg) |
230 | 243 | unit_test_sim.AddModelToTask(unit_task_name, intertialAttitudeFilter)
|
231 | 244 |
|
232 | 245 | # Add test module to runtime call list
|
@@ -294,7 +307,7 @@ def test_measurements_kf(show_plots, initial_error, method):
|
294 | 307 |
|
295 | 308 | accel_data = messaging.AccDataMsgPayload()
|
296 | 309 | accel_measurement = messaging.AccDataMsg().write(accel_data)
|
297 |
| - intertialAttitudeFilter.accelDataMsg.subscribeTo(accel_measurement) |
| 310 | + miru_low_pass_filter_converter.imuAccelDataInMsg.subscribeTo(accel_measurement) |
298 | 311 |
|
299 | 312 | filter_data_log = intertialAttitudeFilter.inertialFilterOutputMsg.recorder()
|
300 | 313 | unit_test_sim.AddModelToTask(unit_task_name, filter_data_log)
|
|
0 commit comments