Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#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_ */