Tempus_UnitTest_ERK_General.cpp

Go to the documentation of this file.

//@HEADER
// *****************************************************************************
//          Tempus: Time Integration and Sensitivity Analysis Package
//
// Copyright 2017 NTESS and the Tempus contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
//@HEADER

#include "Tempus_UnitTest_RK_Utils.hpp"

namespace Tempus_Unit_Test {

using Teuchos::RCP;
using Teuchos::rcp;
using Teuchos::rcp_const_cast;
using Teuchos::rcp_dynamic_cast;

// ************************************************************
// ************************************************************
TEUCHOS_UNIT_TEST(ERK_General, Default_Construction)
{
  auto model = rcp(new Tempus_Test::SinCosModel<double>());

  // Default construction.
  auto stepper = rcp(new Tempus::StepperERK_General<double>());
  stepper->setModel(model);
  stepper->initialize();
  TEUCHOS_TEST_FOR_EXCEPT(!stepper->isInitialized());

  // Default values for construction.
  auto modifier             = rcp(new Tempus::StepperRKModifierDefault<double>());
  auto modifierX            = rcp(new Tempus::StepperRKModifierXDefault<double>());
  auto observer             = rcp(new Tempus::StepperRKObserverDefault<double>());
  bool useFSAL              = stepper->getUseFSAL();
  std::string ICConsistency = stepper->getICConsistency();
  bool ICConsistencyCheck   = stepper->getICConsistencyCheck();
  bool useEmbedded          = stepper->getUseEmbedded();

  int NumStages = 4;
  Teuchos::SerialDenseMatrix<int, double> A(NumStages, NumStages);
  Teuchos::SerialDenseVector<int, double> b(NumStages);
  Teuchos::SerialDenseVector<int, double> c(NumStages);
  Teuchos::SerialDenseVector<int, double> bstar(0);

  // Fill A:
  A(0, 0) = 0.0;
  A(0, 1) = 0.0;
  A(0, 2) = 0.0;
  A(0, 3) = 0.0;
  A(1, 0) = 0.5;
  A(1, 1) = 0.0;
  A(1, 2) = 0.0;
  A(1, 3) = 0.0;
  A(2, 0) = 0.0;
  A(2, 1) = 0.5;
  A(2, 2) = 0.0;
  A(2, 3) = 0.0;
  A(3, 0) = 0.0;
  A(3, 1) = 0.0;
  A(3, 2) = 1.0;
  A(3, 3) = 0.0;

  // Fill b:
  b(0) = 1.0 / 6.0;
  b(1) = 1.0 / 3.0;
  b(2) = 1.0 / 3.0;
  b(3) = 1.0 / 6.0;

  // fill c:
  c(0) = 0.0;
  c(1) = 0.5;
  c(2) = 0.5;
  c(3) = 1.0;

  int order = 4;

  // Test the set functions.
  stepper->setAppAction(modifier);
  stepper->initialize();
  TEUCHOS_TEST_FOR_EXCEPT(!stepper->isInitialized());
  stepper->setAppAction(modifierX);
  stepper->initialize();
  TEUCHOS_TEST_FOR_EXCEPT(!stepper->isInitialized());
  stepper->setAppAction(observer);
  stepper->initialize();
  TEUCHOS_TEST_FOR_EXCEPT(!stepper->isInitialized());
  stepper->setUseFSAL(useFSAL);
  stepper->initialize();
  TEUCHOS_TEST_FOR_EXCEPT(!stepper->isInitialized());
  stepper->setICConsistency(ICConsistency);
  stepper->initialize();
  TEUCHOS_TEST_FOR_EXCEPT(!stepper->isInitialized());
  stepper->setICConsistencyCheck(ICConsistencyCheck);
  stepper->initialize();
  TEUCHOS_TEST_FOR_EXCEPT(!stepper->isInitialized());
  stepper->setUseEmbedded(useEmbedded);
  stepper->initialize();
  TEUCHOS_TEST_FOR_EXCEPT(!stepper->isInitialized());

  stepper->setTableau(A, b, c, order, order, order);
  stepper->initialize();
  TEUCHOS_TEST_FOR_EXCEPT(!stepper->isInitialized());

  // Full argument list construction.
  stepper = rcp(new Tempus::StepperERK_General<double>(
      model, useFSAL, ICConsistency, ICConsistencyCheck, useEmbedded, A, b, c,
      order, order, order, bstar, modifier));
  TEUCHOS_TEST_FOR_EXCEPT(!stepper->isInitialized());

  // Test stepper properties.
  TEUCHOS_ASSERT(stepper->getOrder() == 4);
}

// ************************************************************
// ************************************************************
TEUCHOS_UNIT_TEST(ERK_General, StepperFactory_Construction)
{
  auto model = rcp(new Tempus_Test::SinCosModel<double>());
  testFactoryConstruction("General ERK", model);
}

// ************************************************************
// ************************************************************
TEUCHOS_UNIT_TEST(ERK_General, AppAction)
{
  auto stepper = rcp(new Tempus::StepperERK_General<double>());
  auto model   = rcp(new Tempus_Test::SinCosModel<double>());
  testRKAppAction(stepper, model, out, success);
}

}  // namespace Tempus_Unit_Test