13#include "../TestModels/DahlquistTestModel.hpp"
20using Teuchos::rcp_const_cast;
21using Teuchos::rcp_dynamic_cast;
22using Teuchos::ParameterList;
23using Teuchos::sublist;
34 TEUCHOS_ASSERT(stepper->getOrder() == 2);
59class StepperRKModifierSDIRK21Test
65 StepperRKModifierSDIRK21Test(Teuchos::FancyOStream &Out,
bool &Success)
66 : out(Out), success(Success)
70 virtual ~StepperRKModifierSDIRK21Test(){}
78 const double relTol = 1.0e-14;
79 auto stageNumber = stepper->getStageNumber();
80 Teuchos::SerialDenseVector<int,double> c = stepper->getTableau()->c();
82 auto currentState = sh->getCurrentState();
83 auto workingState = sh->getWorkingState();
84 const double dt = workingState->getTimeStep();
85 double time = currentState->getTime();
86 if (stageNumber >= 0) time += c(stageNumber)*dt;
88 auto x = workingState->getX();
89 auto xDot = workingState->getXDot();
90 if (xDot == Teuchos::null) xDot = stepper->getStepperXDot();
93 case StepperRKAppAction<double>::BEGIN_STEP:
96 auto DME = Teuchos::rcp_dynamic_cast<
98 TEST_FLOATING_EQUALITY(DME->getLambda(), -1.0, relTol);
100 TEST_FLOATING_EQUALITY(dt, 1.0, relTol);
102 const double x_0 = get_ele(*(x), 0);
103 const double xDot_0 = get_ele(*(xDot), 0);
104 TEST_FLOATING_EQUALITY(x_0, 1.0, relTol);
105 TEST_FLOATING_EQUALITY(xDot_0, -1.0, relTol);
106 TEST_ASSERT(std::abs(time) < relTol);
107 TEST_FLOATING_EQUALITY(dt, 1.0, relTol);
108 TEST_COMPARE(stageNumber, ==, -1);
111 case StepperRKAppAction<double>::BEGIN_STAGE:
112 case StepperRKAppAction<double>::BEFORE_SOLVE:
114 const double X_i = get_ele(*(x), 0);
115 const double f_i = get_ele(*(xDot), 0);
117 if (stageNumber == 0) {
118 TEST_FLOATING_EQUALITY(X_i, 1.0, relTol);
119 TEST_FLOATING_EQUALITY(time, 1.0, relTol);
120 TEST_ASSERT(std::abs(f_i) < relTol);
121 }
else if (stageNumber == 1) {
122 TEST_FLOATING_EQUALITY(X_i, 0.5, relTol);
123 TEST_FLOATING_EQUALITY(f_i, -1.0, relTol);
124 TEST_ASSERT(std::abs(time) < relTol);
126 TEUCHOS_TEST_FOR_EXCEPT( !(-1 < stageNumber && stageNumber < 2));
131 case StepperRKAppAction<double>::AFTER_SOLVE:
132 case StepperRKAppAction<double>::BEFORE_EXPLICIT_EVAL:
133 case StepperRKAppAction<double>::END_STAGE:
135 const double X_i = get_ele(*(x), 0);
136 const double f_i = get_ele(*(xDot), 0);
138 if (stageNumber == 0) {
140 TEST_FLOATING_EQUALITY(X_i, 0.5, relTol);
141 TEST_FLOATING_EQUALITY(time, 1.0, relTol);
142 TEST_FLOATING_EQUALITY(f_i, -0.5, relTol);
143 }
else if (stageNumber == 1) {
145 TEST_FLOATING_EQUALITY(X_i, 0.75, relTol);
146 TEST_FLOATING_EQUALITY(f_i, -0.75, relTol);
147 TEST_ASSERT(std::abs(time) < relTol);
149 TEUCHOS_TEST_FOR_EXCEPT( !(-1 < stageNumber && stageNumber < 2));
154 case StepperRKAppAction<double>::END_STEP:
156 const double x_1 = get_ele(*(x), 0);
157 time = workingState->getTime();
158 TEST_FLOATING_EQUALITY(x_1, 3.0/8.0, relTol);
159 TEST_FLOATING_EQUALITY(time, 1.0, relTol);
160 TEST_FLOATING_EQUALITY(dt, 1.0, relTol);
161 TEST_COMPARE(stageNumber, ==, -1);
163 if (stepper->getUseEmbedded() ==
true) {
164 TEST_FLOATING_EQUALITY(workingState->getTolRel(), 1.0, relTol);
165 TEST_ASSERT(std::abs(workingState->getTolAbs()) < relTol);
167 TEST_ASSERT(std::abs(workingState->getErrorRel()) < relTol);
179 Teuchos::FancyOStream & out;
189 Teuchos::RCP<const Thyra::ModelEvaluator<double> >
192 auto modifier = rcp(
new StepperRKModifierSDIRK21Test(out, success));
194 stepper->setModel(model);
195 stepper->setAppAction(modifier);
196 stepper->setICConsistency(
"Consistent");
197 stepper->setUseFSAL(
false);
198 stepper->initialize();
204 stepper->setInitialConditions(solutionHistory);
205 solutionHistory->initWorkingState();
207 solutionHistory->getWorkingState()->setTimeStep(dt);
208 solutionHistory->getWorkingState()->setTime(dt);
209 stepper->takeStep(solutionHistory);
212 TEUCHOS_ASSERT(stepper->getOrder() == 2);
SolutionHistory is basically a container of SolutionStates. SolutionHistory maintains a collection of...
ACTION_LOCATION
Indicates the location of application action (see algorithm).
Base class for Runge-Kutta methods, ExplicitRK, DIRK and IMEX.
Base modifier for StepperRK.
The classic Dahlquist Test Problem.
Sine-Cosine model problem from Rythmos. This is a canonical Sine-Cosine differential equation.
void testDIRKAccessorsFullConstruction(const RCP< Tempus::StepperDIRK< double > > &stepper)
Unit test utility for ExplicitRK Stepper construction and accessors.
void testRKAppAction(const Teuchos::RCP< Tempus::StepperRKBase< double > > &stepper, const Teuchos::RCP< const Thyra::ModelEvaluator< double > > &model, Teuchos::FancyOStream &out, bool &success)
Unit test utility for Stepper RK AppAction.
TEUCHOS_UNIT_TEST(BackwardEuler, Default_Construction)
void testFactoryConstruction(std::string stepperType, const Teuchos::RCP< const Thyra::ModelEvaluator< double > > &model)
Unit test utility for Stepper construction through StepperFactory.
Teuchos::RCP< SolutionHistory< Scalar > > createSolutionHistoryME(const Teuchos::RCP< const Thyra::ModelEvaluator< Scalar > > &model)
Nonmember contructor from a Thyra ModelEvaluator.