step/trustregion/test_01.cpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//               Rapid Optimization Library (ROL) Package
//
// Copyright 2014 NTESS and the ROL contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#include "ROL_OptimizationSolver.hpp"
#include "Teuchos_GlobalMPISession.hpp"

#include "ROL_Stream.hpp"
#include "ROL_RandomVector.hpp"
#include "ROL_Rosenbrock.hpp"
#include "ROL_ConicApproximationModel.hpp"

#include <iostream>


int main(int argc, char *argv[]) {

  using RealT = double;
  using namespace ROL;

  Teuchos::GlobalMPISession mpiSession(&argc, &argv);

  // This little trick lets us print to std::cout only if a (dummy) command-line argument is provided.
  auto outStream = makeStreamPtr( std::cout, argc > 1 );

  int errorFlag  = 0;

  // *** Test body.

  try {

    ZOO::getRosenbrock<RealT> rosenbrock;

    auto x   = rosenbrock.getInitialGuess();
    auto a   = x->clone();
    auto s   = x->clone();
    auto d   = x->clone();
    auto obj = rosenbrock.getObjective();

    RandomizeVector( *s );
    RandomizeVector( *a );
    RandomizeVector( *d );

    ConicApproximationModel<RealT> conic( obj, x, s, a );
    conic.checkGradient( *x, *d, true, *outStream );

  }
  catch (std::logic_error& err) {
    *outStream << err.what() << std::endl;
    errorFlag = -1000;
  }; // end try

  if (errorFlag != 0)
    std::cout << "End Result: TEST FAILED" << std::endl;
  else
    std::cout << "End Result: TEST PASSED" << std::endl;

  return 0;

}