testMinou.cc

Go to the documentation of this file.

#include <cppunit/extensions/HelperMacros.h>
#include <tst/Verify.h>
#include <utl/Minou.h>

using namespace utl;
using namespace std;
using namespace tst;


#define EQUAL(x, y) CPPUNIT_ASSERT(Verify<Equal>(x, y))


struct MyFunc : public Minou::Base {

  MyFunc() : Base({{"x", 2, 0.1}}, 0.5) { }

  double operator()(const vector<double>& pars) const
  { return utl::Sqr(pars.front() - 1); }

};


struct HenonHeiles : public Minou::Base {

  HenonHeiles() : Base({{"x", 2, 0.1}, {"y", -1, 0.1}}, 0.5) { }

  double
  operator()(const vector<double>& pars)
     const
  {
    const double& x = pars[0];
    const double& y = pars[1];
    return 0.5*(Sqr(x) + Sqr(y)) + Sqr(x)*y - (1/3.)*Sqr(y)*y;
  }

};


struct Polynomial : public Minou::Base {

  Polynomial(const unsigned int n) { GetParameterDefs() = GenerateParameters(n); }

  vector<Minou::ParameterDef>
  GenerateParameters(const unsigned int n)
  {
    vector<Minou::ParameterDef> p;
    for (unsigned int i = 0; i <= n; ++i)
      p.emplace_back(std::string("a") + std::to_string(i), 0, 1);
    return p;
  }

  double operator()(const vector<double>& pars) { return pars.at(0); }

};


class TestMinou : public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(TestMinou);
  CPPUNIT_TEST(TestMyFunc);
  CPPUNIT_TEST(TestHenonHeiles);
  CPPUNIT_TEST(TestPolynomial);
  CPPUNIT_TEST_SUITE_END();

public:
  void setUp() { }

  void tearDown() { }

  void
  TestMyFunc()
  {
    MyFunc func;
    Minou::Minimizer<MyFunc> mini(func);
    CPPUNIT_ASSERT(!mini.Minimize());
    const auto x = mini.GetParameterAndError(0);
    cout << x.first << ' ' << x.second << endl;
  }

  void
  TestHenonHeiles()
  {
    HenonHeiles func;
    Minou::Minimizer<HenonHeiles> mini(func);
    if (mini.Minimize()) {
      cout << "ok: failed first time" << endl;
      func.GetParameterDefs()[0].fValue = 0.1;
      func.GetParameterDefs()[1].fValue = -0.1;
      mini.DefineParameters();
      CPPUNIT_ASSERT(!mini.Minimize());
    }
    double x, ex;
    mini.GetMinuit().GetParameter(0, x, ex);
    double y, ey;
    mini.GetMinuit().GetParameter(1, y, ey);
    cout
      << x << ' ' << ex << ' '
      << y << ' ' << ey
      << endl;
  }

  void
  TestPolynomial()
  {
    Polynomial p(10);
    Minou::Minimizer<Polynomial> mini(p);
  }

};


CPPUNIT_TEST_SUITE_REGISTRATION(TestMinou);