Add assertion for hashFunction in BucketedRandomProjectionLSHSuite

apache · Nov 14, 2016 · adbbefe · adbbefe
1 parent 0e9250b
commit adbbefe
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mllib/src/test/scala/org/apache/spark/ml/feature/BucketedRandomProjectionLSHSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/feature/BucketedRandomProjectionLSHSuite.scala
@@ -77,6 +77,7 @@ class BucketedRandomProjectionLSHSuite
     model.set(model.numHashFunctions, 2)
     val res = model.hashFunction(Vectors.dense(1.23, 4.56))
     assert(res(0).equals(Vectors.dense(9.0, 2.0)))
+    assert(res(1).equals(Vectors.dense(7.0, 4.0)))
   }
 
   test("keyDistance") {