int main()
{
LeastSquares test;
// test simple uniform weight
test.add(1,1);
test.add(2,2);
test.add(3,3);
test.add(4,4);
test.add(5,5);
test.add(6,6);
LeastSquares::Linear result = test.fitLinearVertical();
SINVARIANT(result.intercept == 0 && result.slope == 1);
SINVARIANT(Double::eq(result.get(113), 113));
SINVARIANT(test.size() == 6);
// test simple non-uniform weight
test.clear();
test.add(-1,0, 0.25);
test.add(0,1, 0.50);
test.add(1,0, 0.25);
result = test.fitLinearVertical();
// compare against hand calculation
SINVARIANT(result.intercept == 0.5 && result.slope == 0);
// test complex uniform weight
test.clear();
test.add(24037007,-321.5);
test.add(27040842,-322);
test.add(28042496,-323);
test.add(29043391,-314);
test.add(31044518,-322);
test.add(39048310,-311);
test.add(62067122,-317.5);
test.add(116106946,-300);
test.add(117107339,-301.5);
test.add(118108081,-290);
test.add(119109345,-296);
test.add(120110256,-294.5);
test.add(122113491,-289);
test.add(125114789,-293.5);
test.add(126115896,-296.5);
test.add(127116738,-292.5);
test.add(128116805,-283);
test.add(129116889,-285);
test.add(130116954,-281);
test.add(131117031,-279);
test.add(132117104,-279);
test.add(133117182,-278);
test.add(134117254,-273);
test.add(135117331,-275.5);
test.add(136118414,-278.5);
test.add(137119481,-275);
test.add(138119556,-274.5);
test.add(139119633,-276);
test.add(140119706,-273.5);
test.add(141119784,-273);
test.add(142119857,-273);
test.add(143119954,-276);
test.add(144120009,-268);
test.add(145120085,-269);
test.add(146120159,-269.5);
test.add(147120235,-271);
test.add(148120311,-267.5);
test.add(149120387,-269);
test.add(150120469,-269.5);
result = test.fitLinearVertical();
// Results calculated using perl Statistics::OLS module
INVARIANT(Double::eq(result.intercept,-334.874623780947) &&
Double::eq(result.slope, 4.10303516778801e-07), "bad");
// test complex uniform weight
// the following numbers are manually randomly typed in
test.clear();
test.add(23, -3.5, 0.1);
test.add(-2, 10, 0.05);
test.add(2, 30.5, 0.3);
test.add(100, 45.3, 1.2);
test.add(26.4, 75.3, 2.0);
test.add(4.3, 57.2, 1.0);
test.add(24, 23.0, 3.0);
test.add(53, -40.2, 0.1);
test.add(134, 1034.2, 0.56);
test.add(5234, -304.56, 1.23);
test.add(1345, -201.34, 3.24);
test.add(21, 1123, 7.96);
test.add(1, 367.34, 3.22);
test.add(1345, 560.345, 0.32);
test.add(5678, 272.101, 7.11);
test.add(7890, 909.1, 8.09);
test.add(4245, 232.5, 10.01);
test.add(2490, 934, 21.89);
test.add(345, 10293.3, 0.005);
test.add(34, 984, 7.36);
result = test.fitLinearVertical();
// Results calculated using perl Statistics::LineFit module
INVARIANT(Double::eq(result.intercept, 662.946833319873) &&
Double::eq(result.slope, -0.0124699082339257), "bad");
cout << "Passed least_squares test\n";
return 0;
}
