ExternalVendorCode/trick
1
0
Fork 0
mirror of https://github.com/nasa/trick.git synced 2024-12-18 20:57:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

SAIntegrator: Add RKF45Integrator_unittest. #1114

Browse Source
This commit is contained in:
Penn, John M 047828115 2021-03-02 13:23:26 -06:00
parent a4df0bddcb
commit a5e67ae3a9
2 changed files with 89 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
trick_source/trick_utils/SAIntegrator/unittest/Makefile
View File

@ -20,6 +20,7 @@ TEST_PROGRAMS = SAIntegrator_unittest\
RK2Integrator_unittest\
RK4Integrator_unittest\
RK3_8Integrator_unittest\
RKF45Integrator_unittest\
RootFinder_unittest
all: test
@ -32,6 +33,7 @@ test: $(TEST_PROGRAMS)
./RK2Integrator_unittest --gtest_output=xml:${TRICK_HOME}/trick_test/RK2Integrator_unittest.xml
./RK4Integrator_unittest --gtest_output=xml:${TRICK_HOME}/trick_test/RK4Integrator_unittest.xml
./RK3_8Integrator_unittest --gtest_output=xml:${TRICK_HOME}/trick_test/RK3_8Integrator_unittest.xml
./RKF45Integrator_unittest --gtest_output=xml:${TRICK_HOME}/trick_test/RKF45Integrator_unittest.xml
./RootFinder_unittest --gtest_output=xml:${TRICK_HOME}/trick_test/RootFinder_unittest.xml
SAIntegrator_unittest.o : SAIntegrator_unittest.cc
@ -76,6 +78,12 @@ RK3_8Integrator_unittest.o : RK3_8Integrator_unittest.cc
RK3_8Integrator_unittest : ${SAI_LIBDIR}/${SAI_LIBNAME} RK3_8Integrator_unittest.o
$(TRICK_CXX) $(TRICK_CPPFLAGS) -o $@ $^ ${LIBDIRS} -lSAInteg -lgtest -lgtest_main -lpthread
# ====
RKF45Integrator_unittest.o : RKF45Integrator_unittest.cc
$(TRICK_CXX) $(TRICK_CPPFLAGS) $(INCLUDE_DIRS) -c $<
RKF45Integrator_unittest : ${SAI_LIBDIR}/${SAI_LIBNAME} RKF45Integrator_unittest.o
$(TRICK_CXX) $(TRICK_CPPFLAGS) -o $@ $^ ${LIBDIRS} -lSAInteg -lgtest -lgtest_main -lpthread
# ====
RootFinder_unittest.o : RootFinder_unittest.cc
$(TRICK_CXX) $(TRICK_CPPFLAGS) $(INCLUDE_DIRS) -c $<
@ -96,4 +104,5 @@ spotless: clean
${RM} RK2Integrator_unittest
${RM} RK4Integrator_unittest
${RM} RK3_8Integrator_unittest
${RM} RKF45Integrator_unittest
${RM} RootFinder_unittest

80
trick_source/trick_utils/SAIntegrator/unittest/RKF45Integrator_unittest.cc Normal file
View File

@ -0,0 +1,80 @@
#include <gtest/gtest.h>
#include <iostream>
#include "SAIntegrator.hh"
#include <math.h>
#define EXCEPTABLE_GLOBAL_ERROR 0.00001
#define EXCEPTABLE_LOCAL_ERROR 0.000001
/* ========================================================
RKF4(5) Basic Test
Problem: y' = y - x^2 + 1
y(0) = 0.5
for 0 <= x <= 2
Solution: y = x^2 + 2*x + 1 - 0.5*e^x ; where x = 2.
y = 5.30547.
========================================================
*/
void drv( double x, double state[], double derivs[], void* udata __attribute__((unused))) {
double x2 = x*x;
derivs[0] = state[0] - x2 + 1;
}
TEST(RKF45Integrator_unittest, basic_test) {
double lower_limit = 0.0;
double upper_limit = 2.0;
double result;
double h = 0.2;
double* state[1] = { &result };
SA::RKF45Integrator integ(EXCEPTABLE_LOCAL_ERROR, h, 1, state, drv, NULL);
result = 0.5;
integ.setIndyVar(lower_limit);
double x = lower_limit;
while (x < upper_limit) {
integ.load();
h = integ.adaptive_step(h);
integ.unload();
x = integ.getIndyVar();
h = fmin(h,(upper_limit-x));
}
EXPECT_NEAR(result, 5.30547, EXCEPTABLE_GLOBAL_ERROR);
}
/* ========================================================
Multi-Variable Test
========================================================
*/
void deriv4( double t,
double state[] __attribute__((unused)),
double derivs[],
void* udata __attribute__((unused))) {
double t2 = t*t;
double t3 = t2*t;
derivs[0] = 3.0*t3 - 3.0*t2 - 3.0*t + 4.0;
derivs[1] = 2.0*t3 + 1.0*t2 - 5.0*t + 7.0;
derivs[2] = t3 + t2 + 5.0*t + 4.0;
derivs[3] = t3 - 6.0*t2 + 4.0*t + 12.0;
}
TEST(RKF45Integrator_unittest, multi_var_test) {
double lower_limit = 0.0;
double upper_limit = 2.0;
double state[4] = {0.0, 0.0, 0.0, 0.0};
double* state_var_p[4] = { &(state[0]), &(state[1]), &(state[2]), &(state[3]) };
double h = 0.2;
SA::RKF45Integrator integ(EXCEPTABLE_LOCAL_ERROR, h, 4, state_var_p, deriv4, NULL);
integ.setIndyVar(lower_limit);
double x = lower_limit;
while (x < upper_limit) {
integ.load();
h = integ.adaptive_step(h);
integ.unload();
x = integ.getIndyVar();
h = fmin(h,(upper_limit-x));
}
EXPECT_NEAR(state[0], 6.0, EXCEPTABLE_GLOBAL_ERROR);
EXPECT_NEAR(state[1], (14.0+2/3.0), EXCEPTABLE_GLOBAL_ERROR);
EXPECT_NEAR(state[2], (24+2/3.0), EXCEPTABLE_GLOBAL_ERROR);
EXPECT_NEAR(state[3], 20.0, EXCEPTABLE_GLOBAL_ERROR);
}