check that remaining prediction is equal

Nick Pentreath · Nick Pentreath · commit 041a3b3fa3c9 · 2017-02-27T12:29:15.000+02:00
diff --git a/mllib/src/test/scala/org/apache/spark/ml/recommendation/ALSSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/recommendation/ALSSuite.scala
@@ -581,6 +581,7 @@ class ALSSuite
       .select("prediction").as[Float].collect()
     assert(dropPredictions.length == 1)
     assert(!dropPredictions.head.isNaN)
+    assert(dropPredictions.head ~== defaultPredictions.last relTol 1e-14)
   }
 
   test("case insensitive cold start param value") {

Original file line number	Diff line number	Diff line change
`@@ -581,6 +581,7 @@ class ALSSuite`
`581`	`581`	`.select("prediction").as[Float].collect()`
`582`	`582`	`assert(dropPredictions.length == 1)`
`583`	`583`	`assert(!dropPredictions.head.isNaN)`
	`584`	`+ assert(dropPredictions.head ~== defaultPredictions.last relTol 1e-14)`
`584`	`585`	`}`
`585`	`586`
`586`	`587`	`test("case insensitive cold start param value") {`