Source code for test_tempMeasurement

# 
#  ISC License
# 
#  Copyright (c) 2023, Autonomous Vehicle Systems Lab, University of Colorado Boulder
# 
#  Permission to use, copy, modify, and/or distribute this software for any
#  purpose with or without fee is hereby granted, provided that the above
#  copyright notice and this permission notice appear in all copies.
# 
#  THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
#  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
#  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
#  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
#  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
#  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
#  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
# 
# 

#
# Thermal Sensor Faults Test
#
# Purpose:  Test the proper function of the faulty behaviors of the tempMeasurement module.
#           Results are compared to simple truth values.
#           To test spiking, percent of spike values compared with expected.
# Creation Date:  Feb. 9, 2023
#

import os

import numpy as np
import pytest
from Basilisk.architecture import messaging
from Basilisk.simulation import sensorThermal, tempMeasurement
from Basilisk.utilities import SimulationBaseClass
from Basilisk.utilities import macros, astroFunctions
from Basilisk.utilities import unitTestSupport

path = os.path.dirname(os.path.abspath(__file__))

# The following 'parametrize' function decorator provides the parameters and expected results for each
#   of the multiple test runs for this test.
# Need RNGSeed 464374481
[docs]@pytest.mark.parametrize( "tempFault", [ "TEMP_FAULT_NOMINAL", "TEMP_FAULT_STUCK_CURRENT", "TEMP_FAULT_STUCK_VALUE", "TEMP_FAULT_SPIKING", "BIASED" ]) # provide a unique test method name, starting with test_ def test_sensorThermalFault(tempFault): '''This function is called by the py.test environment.''' # each test method requires a single assert method to be called [testResults, testMessage] = run(tempFault) assert testResults < 1, testMessage __tracebackhide__ = True
def run(tempFault): testFailCount = 0 # zero unit test result counter testMessages = [] # create empty list to store test log messages # Create simulation variable names unitTaskName = "unitTask" # arbitrary name (don't change) unitProcessName = "TestProcess" # arbitrary name (don't change) # Create a sim module as an empty container unitTestSim = SimulationBaseClass.SimBaseClass() # set the simulation time variable used later on simulationTime = macros.sec2nano(100.) # # create the simulation process # testProcessRate = macros.sec2nano(1.0) # update process rate update time testProc = unitTestSim.CreateNewProcess(unitProcessName) testProc.addTask(unitTestSim.CreateNewTask(unitTaskName, testProcessRate)) # # set up the simulation tasks/objects # # set device status message sensorStatusMsgPayload = messaging.DeviceStatusMsgPayload() sensorStatusMsgPayload.deviceStatus = 1 sensorStatusMsg = messaging.DeviceStatusMsg().write(sensorStatusMsgPayload) # set the spacecraft message scStateMsgPayload = messaging.SCStatesMsgPayload() scStateMsgPayload.r_BN_N = [6378*1000., 0., 0.] scStateMsgPayload.sigma_BN = [0., 0., 0.] scStateMsg = messaging.SCStatesMsg().write(scStateMsgPayload) # set the sun message sunMsgPayload = messaging.SpicePlanetStateMsgPayload() sunMsgPayload.PositionVector = [astroFunctions.AU*1000., 0., 0.] sunMsg = messaging.SpicePlanetStateMsg().write(sunMsgPayload) # # set up the truth value temperature modeling # sensorThermalModel = sensorThermal.SensorThermal() sensorThermalModel.ModelTag = 'sensorThermalModel' sensorThermalModel.nHat_B = [0, 0, 1] sensorThermalModel.sensorArea = 1.0 # m^2 sensorThermalModel.sensorAbsorptivity = 0.25 sensorThermalModel.sensorEmissivity = 0.34 sensorThermalModel.sensorMass = 2.0 # kg sensorThermalModel.sensorSpecificHeat = 890 sensorThermalModel.sensorPowerDraw = 30.0 # W sensorThermalModel.T_0 = 0 # [ÂșC] sensorThermalModel.sunInMsg.subscribeTo(sunMsg) sensorThermalModel.stateInMsg.subscribeTo(scStateMsg) sensorThermalModel.sensorStatusInMsg.subscribeTo(sensorStatusMsg) unitTestSim.AddModelToTask(unitTaskName, sensorThermalModel) # # set up the tempMeasurement module # tempMeasurementModel = tempMeasurement.TempMeasurement() tempMeasurementModel.tempInMsg.subscribeTo(sensorThermalModel.temperatureOutMsg) unitTestSim.AddModelToTask(unitTaskName, tempMeasurementModel) # # log data # # log the RW temperature tempLog = tempMeasurementModel.tempOutMsg.recorder() unitTestSim.AddModelToTask(unitTaskName, tempLog) # # initialize Simulation # # Truth Values if tempFault == "TEMP_FAULT_NOMINAL": truthValue = -4.251289338192501 unitTestSim.InitializeSimulation() unitTestSim.ConfigureStopTime(simulationTime) unitTestSim.ExecuteSimulation() elif tempFault == "TEMP_FAULT_STUCK_CURRENT": truthValue = -2.1722164230619447 unitTestSim.InitializeSimulation() unitTestSim.ConfigureStopTime(round(simulationTime/2.0)) unitTestSim.ExecuteSimulation() tempMeasurementModel.faultState = tempMeasurement.TEMP_FAULT_STUCK_CURRENT unitTestSim.ConfigureStopTime(simulationTime) unitTestSim.ExecuteSimulation() elif tempFault == "TEMP_FAULT_STUCK_VALUE": truthValue = 10.0 tempMeasurementModel.stuckValue = 10.0 unitTestSim.InitializeSimulation() unitTestSim.ConfigureStopTime(round(simulationTime/2.0)) unitTestSim.ExecuteSimulation() tempMeasurementModel.faultState = tempMeasurement.TEMP_FAULT_STUCK_VALUE unitTestSim.ConfigureStopTime(simulationTime) unitTestSim.ExecuteSimulation() elif tempFault == "TEMP_FAULT_SPIKING": tempMeasurementModel.faultState = tempMeasurement.TEMP_FAULT_SPIKING truthValue = 0.1 unitTestSim.InitializeSimulation() unitTestSim.ConfigureStopTime(simulationTime) unitTestSim.ExecuteSimulation() nominals = np.array([ 0., -0.04439869, -0.08875756, -0.13307667, -0.17735608, -0.22159584, -0.265796, -0.30995662, -0.35407775, -0.39815946, -0.44220178, -0.48620479, -0.53016852, -0.57409304, -0.6179784, -0.66182465, -0.70563184, -0.74940004, -0.79312929, -0.83681965, -0.88047117, -0.9240839, -0.9676579, -1.01119321, -1.0546899, -1.09814802, -1.14156761, -1.18494873, -1.22829144, -1.27159578, -1.31486181, -1.35808958, -1.40127914, -1.44443055, -1.48754385, -1.5306191, -1.57365634, -1.61665564, -1.65961704, -1.7025406, -1.74542636, -1.78827437, -1.83108469, -1.87385737, -1.91659246, -1.95929001, -2.00195007, -2.04457269, -2.08715792, -2.12970582, -2.17221642, -2.21468979, -2.25712597, -2.29952502, -2.34188697, -2.38421189, -2.42649982, -2.46875082, -2.51096492, -2.55314219, -2.59528266, -2.6373864, -2.67945344, -2.72148384, -2.76347765, -2.80543491, -2.84735568, -2.88924, -2.93108792, -2.9728995 , -3.01467477, -3.05641378, -3.0981166, -3.13978325, -3.1814138, -3.22300829, -3.26456677, -3.30608928, -3.34757587, -3.38902659, -3.43044149, -3.47182062, -3.51316402, -3.55447174, -3.59574383, -3.63698033, -3.67818129, -3.71934677, -3.76047679, -3.80157142, -3.8426307, -3.88365467, -3.92464338, -3.96559689, -4.00651522, -4.04739844, -4.08824658, -4.1290597, -4.16983783, -4.21058103, -4.25128934]) testValue = 0 sensorTemp = np.array(tempLog.temperature) for ii in range(len(nominals)): if abs(nominals[ii]) > 0.0001: if abs(nominals[ii] - sensorTemp[ii]) > 1e-6: testValue = testValue + 1 else: if abs(sensorTemp[ii]) > 1e-6: testValue = testValue + 1 testValue = testValue/len(nominals) elif tempFault == "BIASED": biasVal = 1.0 tempMeasurementModel.senBias = biasVal truthValue = -3.251289338192501 unitTestSim.InitializeSimulation() unitTestSim.ConfigureStopTime(simulationTime) unitTestSim.ExecuteSimulation() else: NotImplementedError("Fault type specified does not exist.") sensorTemp = np.array(tempLog.temperature) print(sensorTemp) # # compare the module results to the true values # if tempFault == "TEMP_FAULT_SPIKING": if not unitTestSupport.isDoubleEqualRelative(testValue, truthValue, 5E-1): # only 101 values so need this to be relatively relaxed (within 50%) testFailCount+= 1 elif not unitTestSupport.isDoubleEqualRelative(sensorTemp[-1], truthValue, 1E-12): testFailCount += 1 return [testFailCount, ''.join(testMessages)] # each test method requires a single assert method to be called # this check below just makes sure no sub-test failures were found if __name__ == "__main__": run("TEMP_FAULT_NOMINAL")