fix ut

apache · Dec 19, 2017 · 26fe05e · 26fe05e
1 parent 2ecdc73
commit 26fe05e
Showing 1 changed file with 2 additions and 3 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
@@ -423,11 +423,10 @@ class BucketizerSuite extends SparkFunSuite with MLlibTestSparkContext with Defa
       .setOutputCols(Array("result1", "result2"))
 
     Seq(invalid1, invalid2, invalid3).foreach { bucketizer =>
-      // When both inputCol and inputCols are set, we throw Exception.
-      val e = intercept[IllegalArgumentException] {
+      // When both inputCol/outputCol and inputCols/outputCols are set, we throw Exception.
+      intercept[IllegalArgumentException] {
         bucketizer.transform(df)
       }
-      assert(e.getMessage.contains("Both `inputCol` and `inputCols` are set"))
     }
   }
 }