svn commit: r1405333 - /hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java

[tommaso]

Author: tommaso
Date: Sat Nov  3 13:47:07 2012
New Revision: 1405333

URL: http://svn.apache.org/viewvc?rev=1405333&view=rev
Log:
[HAMA-666] - improving unit test for linear regression

Modified:
    
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java

Modified: 
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java
URL: 
http://svn.apache.org/viewvc/hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java?rev=1405333&r1=1405332&r2=1405333&view=diff
==============================================================================
--- 
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java
 (original)
+++ 
hama/trunk/ml/src/test/java/org/apache/hama/ml/regression/LinearRegressionModelTest.java
 Sat Nov  3 13:47:07 2012
@@ -12,7 +12,7 @@ import static org.junit.Assert.assertEqu
 public class LinearRegressionModelTest {
 
   @Test
-  public void testCorrectCalculation() throws Exception {
+  public void testCorrectCostCalculation() throws Exception {
     LinearRegressionModel linearRegressionModel = new LinearRegressionModel();
     DoubleVector x = new DenseDoubleVector(new double[]{2, 3, 4});
     double y = 1;
@@ -20,4 +20,12 @@ public class LinearRegressionModelTest {
     Double cost = linearRegressionModel.calculateCostForItem(x, y, 2, theta);
     assertEquals("wrong cost calculation for linear regression", 
Double.valueOf(16d), cost);
   }
+
+  @Test
+  public void testCorrectHypothesisCalculation() throws Exception {
+    LinearRegressionModel linearRegressionModel = new LinearRegressionModel();
+    Double hypothesisValue = linearRegressionModel.applyHypothesis(new 
DenseDoubleVector(new double[]{1, 1, 1}),
+            new DenseDoubleVector(new double[]{2, 3, 4}));
+    assertEquals("wrong hypothesis value for linear regression", 
Double.valueOf(9), hypothesisValue);
+  }
 }