Source code for test_ephemNavConverter

#
#   Unit Test Script
#   Module Name:        ephemNavConverter
#   Creation Date:      October 16, 2018
#

import inspect
import os

from Basilisk.architecture import messaging
from Basilisk.fswAlgorithms import ephemNavConverter
from Basilisk.utilities import SimulationBaseClass, unitTestSupport, macros
from Basilisk.utilities import astroFunctions

filename = inspect.getframeinfo(inspect.currentframe()).filename
path = os.path.dirname(os.path.abspath(filename))

[docs] def test_ephemNavConverter(): """ Test ephemNavConverter. """ [testResults, testMessage] = ephemNavConverterTestFunction() assert testResults < 1, testMessage
[docs] def ephemNavConverterTestFunction(): """ Test the ephemNavConverter module. Setup a simulation """ testFailCount = 0 # zero unit test result counter testMessages = [] # create empty array to store test log messages 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() # Create test thread testProcessRate = macros.sec2nano(0.5) # update process rate update time testProc = unitTestSim.CreateNewProcess(unitProcessName) testProc.addTask(unitTestSim.CreateNewTask(unitTaskName, testProcessRate)) # Add a new task to the process # Construct the ephemNavConverter module # Set the names for the input messages ephemNav = ephemNavConverter.ephemNavConverter() # This calls the algContain to setup the selfInit, update, and reset ephemNav.ModelTag = "ephemNavConverter" # Add the module to the task unitTestSim.AddModelToTask(unitTaskName, ephemNav) # Create the input message. inputEphem = messaging.EphemerisMsgPayload() # Get the Earth's position and velocity position, velocity = astroFunctions.Earth_RV(astroFunctions.JulianDate([2018, 10, 16])) inputEphem.r_BdyZero_N = position inputEphem.v_BdyZero_N = velocity inputEphem.timeTag = 1.0 # sec inMsg = messaging.EphemerisMsg().write(inputEphem) ephemNav.ephInMsg.subscribeTo(inMsg) dataLog = ephemNav.stateOutMsg.recorder() unitTestSim.AddModelToTask(unitTaskName, dataLog) # Initialize the simulation unitTestSim.InitializeSimulation() # The result isn't going to change with more time. The module will continue to produce the same result unitTestSim.ConfigureStopTime(testProcessRate) # seconds to stop simulation unitTestSim.ExecuteSimulation() posAcc = 1e1 velAcc = 1e-4 outputR = dataLog.r_BN_N outputV = dataLog.v_BN_N outputTime = dataLog.timeTag trueR = [position, position] trueV = [velocity, velocity] trueTime = [inputEphem.timeTag, inputEphem.timeTag] # At each timestep, make sure the vehicleConfig values haven't changed from the initial values testFailCount, testMessages = unitTestSupport.compareArrayND(trueR, outputR, posAcc, "ephemNavConverter output Position", 2, testFailCount, testMessages) testFailCount, testMessages = unitTestSupport.compareArrayND(trueV, outputV, velAcc, "ephemNavConverter output Velocity", 2, testFailCount, testMessages) testFailCount, testMessages = unitTestSupport.compareDoubleArray(trueTime, outputTime, velAcc, "ephemNavConverter output Time", testFailCount, testMessages) # print out success message if no error were found snippentName = "passFail" if testFailCount == 0: colorText = 'ForestGreen' print("PASSED: " + ephemNav.ModelTag) passedText = r'\textcolor{' + colorText + '}{' + "PASSED" + '}' else: colorText = 'Red' print("Failed: " + ephemNav.ModelTag) passedText = r'\textcolor{' + colorText + '}{' + "Failed" + '}' unitTestSupport.writeTeXSnippet(snippentName, passedText, path) return [testFailCount, ''.join(testMessages)]
if __name__ == '__main__': test_ephemNavConverter()