|
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,6 +108,11 @@ 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)
|
@@ -173,7 +179,7 @@ def test_propagation_kf(show_plots):
|
173 | 179 |
|
174 | 180 | accel_data = messaging.AccDataMsgPayload()
|
175 | 181 | accel_measurement = messaging.AccDataMsg().write(accel_data)
|
176 |
| - intertialAttitudeFilter.accelDataMsg.subscribeTo(accel_measurement) |
| 182 | + miru_low_pass_filter_converter.imuAccelDataInMsg.subscribeTo(accel_measurement) |
177 | 183 |
|
178 | 184 | attitude_data_log = intertialAttitudeFilter.inertialFilterOutputMsg.recorder()
|
179 | 185 | unit_test_sim.AddModelToTask(unit_task_name, attitude_data_log)
|
@@ -226,6 +232,11 @@ def test_measurements_kf(show_plots, initial_error, method):
|
226 | 232 | test_process = unit_test_sim.CreateNewProcess(unit_process_name)
|
227 | 233 | test_process.addTask(unit_test_sim.CreateNewTask(unit_task_name, test_process_rate))
|
228 | 234 |
|
| 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 | + |
229 | 240 | intertialAttitudeFilter = inertialAttitudeUkf.InertialAttitudeUkf(method)
|
230 | 241 | unit_test_sim.AddModelToTask(unit_task_name, intertialAttitudeFilter)
|
231 | 242 |
|
@@ -292,9 +303,13 @@ def test_measurements_kf(show_plots, initial_error, method):
|
292 | 303 | star_tracker2.measurementNoise = [[1e-4, 0, 0], [0,1e-4,0], [0,0,1e-4]]
|
293 | 304 | intertialAttitudeFilter.addStarTrackerInput(star_tracker2)
|
294 | 305 |
|
| 306 | + imu_sensor_msg = messaging.IMUSensorMsgPayload() |
| 307 | + imu_sensor_message = messaging.IMUSensorMsg().write(imu_sensor_msg) |
| 308 | + intertialAttitudeFilter.imuSensorDataInMsg.subscribeTo(imu_sensor_message) |
| 309 | + |
295 | 310 | accel_data = messaging.AccDataMsgPayload()
|
296 | 311 | accel_measurement = messaging.AccDataMsg().write(accel_data)
|
297 |
| - intertialAttitudeFilter.accelDataMsg.subscribeTo(accel_measurement) |
| 312 | + miru_low_pass_filter_converter.imuAccelDataInMsg.subscribeTo(accel_measurement) |
298 | 313 |
|
299 | 314 | filter_data_log = intertialAttitudeFilter.inertialFilterOutputMsg.recorder()
|
300 | 315 | unit_test_sim.AddModelToTask(unit_task_name, filter_data_log)
|
|
0 commit comments