one more bucketizer test

jkbradley · jkbradley · commit 1ca973a5e884 · 2015-05-11T12:44:24.000-07:00
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") {

Original file line number	Diff line number	Diff line change
`@@ -101,6 +101,8 @@ class BucketizerSuite extends FunSuite with MLlibTestSparkContext {`
`101`	`101`	`checkBinarySearch(Array(0.0, 1.0, Double.PositiveInfinity))`
`102`	`102`	`// length 3, with -inf and inf`
`103`	`103`	`checkBinarySearch(Array(Double.NegativeInfinity, 1.0, Double.PositiveInfinity))`
	`104`	`+ // length 4, with -inf and inf`
	`105`	`+ checkBinarySearch(Array(Double.NegativeInfinity, 0.0, 1.0, Double.PositiveInfinity))`
`104`	`106`	`}`
`105`	`107`
`106`	`108`	`test("Binary search correctness in contrast with linear search, on random data") {`