#ifndef MANTID_KERNEL_CHEBYSHEVPOLYFITTEST_H_
#define MANTID_KERNEL_CHEBYSHEVPOLYFITTEST_H_

#include <cxxtest/TestSuite.h>

#include "MantidKernel/Math/ChebyshevPolyFit.h"
#include <boost/assign/list_of.hpp>

using Mantid::Kernel::ChebyshevPolyFit;

class ChebyshevPolyFitTest : public CxxTest::TestSuite {
public:
  // This pair of boilerplate methods prevent the suite being created statically
  // This means the constructor isn't called when running other tests
  static ChebyshevPolyFitTest *createSuite() {
    return new ChebyshevPolyFitTest();
  }
  static void destroySuite(ChebyshevPolyFitTest *suite) { delete suite; }

  void test_case_with_quadratic_polynomial() {
    using namespace boost::assign;
    const std::vector<double> xs = list_of(1.0)(2.0)(3.0)(4.0)(5.0);
    const std::vector<double> ys = list_of(1.0)(4.0)(9.0)(16.0)(25.0);
    const std::vector<double> wgts = list_of(1.0)(0.9)(0.5)(0.2)(0.05);

    ChebyshevPolyFit polyfit(2);
    auto coeffs = polyfit(xs, ys, wgts);

    TS_ASSERT_EQUALS(3, coeffs.size());
    const double delta(1e-8);
    TS_ASSERT_DELTA(11, coeffs[0], delta);
    TS_ASSERT_DELTA(12, coeffs[1], delta);
    TS_ASSERT_DELTA(2, coeffs[2], delta);
  }
};

#endif /* MANTID_KERNEL_CHEBYSHEVPOLYFITTEST_H_ */