/** Checks the results of an OLS regression on sample dataset 1. */
@Test
public void testOLSRegression1a() {
double[][] data = createSampleData1();
double[] result1 = Regression.getOLSRegression(data);
assertEquals(.25680930, result1[0], 0.0000001);
assertEquals(0.72792106, result1[1], 0.0000001);
}
/** Checks the results of an OLS regression on sample dataset 2. */
@Test
public void testOLSRegression2a() {
double[][] data = createSampleData2();
double[] result = Regression.getOLSRegression(data);
assertEquals(53.9729697, result[0], 0.0000001);
assertEquals(-4.1823030, result[1], 0.0000001);
}
/**
* Checks the results of an OLS regression on sample dataset 1 AFTER converting it to an XYSeries.
*/
@Test
public void testOLSRegression1b() {
double[][] data = createSampleData1();
XYSeries series = new XYSeries("Test");
for (int i = 0; i < 11; i++) {
series.add(data[i][0], data[i][1]);
}
XYDataset ds = new XYSeriesCollection(series);
double[] result2 = Regression.getOLSRegression(ds, 0);
assertEquals(.25680930, result2[0], 0.0000001);
assertEquals(0.72792106, result2[1], 0.0000001);
}
/**
* Checks the results of an OLS regression on sample dataset 2 AFTER converting it to an XYSeries.
*/
@Test
public void testOLSRegression2b() {
double[][] data = createSampleData2();
XYSeries series = new XYSeries("Test");
for (int i = 0; i < 10; i++) {
series.add(data[i][0], data[i][1]);
}
XYDataset ds = new XYSeriesCollection(series);
double[] result = Regression.getOLSRegression(ds, 0);
assertEquals(53.9729697, result[0], 0.0000001);
assertEquals(-4.1823030, result[1], 0.0000001);
}
