Skip to content

Commit c09c360

Browse files
authored
Update LinearRegressionTest.java formats
1 parent 51021ad commit c09c360

File tree

1 file changed

+27
-27
lines changed

1 file changed

+27
-27
lines changed

src/test/java/com/thealgorithms/machinelearning/LinearRegressionTest.java

Lines changed: 27 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -7,31 +7,31 @@
77

88
class LinearRegressionTest {
99

10-
@Test
11-
void testLinearRegression() {
12-
ArrayList<Double> dependentX = new ArrayList<>();
13-
ArrayList<Double> independentY = new ArrayList<>();
14-
15-
dependentX.add(1.0);
16-
independentY.add(2.0);
17-
dependentX.add(2.0);
18-
independentY.add(3.0);
19-
dependentX.add(3.0);
20-
independentY.add(4.0);
21-
dependentX.add(4.0);
22-
independentY.add(5.0);
23-
dependentX.add(5.0);
24-
independentY.add(6.0);
25-
26-
// Create LinearRegression object
27-
LinearRegression lr = new LinearRegression(dependentX, independentY);
28-
29-
// Check the slope (m) and intercept (c)
30-
assertEquals(1.0, lr.getM(), 0.001);
31-
assertEquals(1.0, lr.getC(), 0.001);
32-
33-
// Check prediction for X = 6
34-
double predictedY = lr.predictForX(6.0);
35-
assertEquals(7.0, predictedY, 0.001);
36-
}
10+
@Test
11+
void testLinearRegression() {
12+
ArrayList<Double> dependentX = new ArrayList<>();
13+
ArrayList<Double> independentY = new ArrayList<>();
14+
15+
dependentX.add(1.0);
16+
independentY.add(2.0);
17+
dependentX.add(2.0);
18+
independentY.add(3.0);
19+
dependentX.add(3.0);
20+
independentY.add(4.0);
21+
dependentX.add(4.0);
22+
independentY.add(5.0);
23+
dependentX.add(5.0);
24+
independentY.add(6.0);
25+
26+
// Create LinearRegression object
27+
LinearRegression lr = new LinearRegression(dependentX, independentY);
28+
29+
// Check the slope (m) and intercept (c)
30+
assertEquals(1.0, lr.getM(), 0.001);
31+
assertEquals(1.0, lr.getC(), 0.001);
32+
33+
// Check prediction for X = 6
34+
double predictedY = lr.predictForX(6.0);
35+
assertEquals(7.0, predictedY, 0.001);
36+
}
3737
}

0 commit comments

Comments
 (0)