one more bucketizer test

marmbrus · May 11, 2015 · 1ca973a · 1ca973a
1 parent 34f124a
commit 1ca973a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mllib/src/test/scala/org/apache/spark/ml/feature/BucketizerSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/feature/BucketizerSuite.scala
@@ -101,6 +101,8 @@ class BucketizerSuite extends FunSuite with MLlibTestSparkContext {
     checkBinarySearch(Array(0.0, 1.0, Double.PositiveInfinity))
     // length 3, with -inf and inf
     checkBinarySearch(Array(Double.NegativeInfinity, 1.0, Double.PositiveInfinity))
+    // length 4, with -inf and inf
+    checkBinarySearch(Array(Double.NegativeInfinity, 0.0, 1.0, Double.PositiveInfinity))
   }
 
   test("Binary search correctness in contrast with linear search, on random data") {