formatting, nits

Author: smurakozi

mllib/src/test/scala/org/apache/spark/ml/clustering/BisectingKMeansSuite.scala

@@ -22,8 +22,7 @@ import org.apache.spark.ml.param.ParamMap
 import org.apache.spark.ml.util.{DefaultReadWriteTest, MLTest, MLTestingUtils}
 import org.apache.spark.sql.Dataset
 
-class BisectingKMeansSuite
-  extends MLTest with DefaultReadWriteTest {
+class BisectingKMeansSuite extends MLTest with DefaultReadWriteTest {
 
   import Encoders._
 
@@ -106,6 +105,7 @@ class BisectingKMeansSuite
     testTransformerByGlobalCheckFunc[Vector](dataset.toDF(), model,
       "features", predictionColName) { rows =>
       val clusters = rows.map(_.getAs[Int](predictionColName)).toSet
+      assert(clusters.size === k)
       assert(clusters === Set(0, 1, 2, 3, 4))
       assert(model.computeCost(dataset) < 0.1)
       assert(model.hasParent)

mllib/src/test/scala/org/apache/spark/ml/clustering/GaussianMixtureSuite.scala

@@ -26,8 +26,7 @@ import org.apache.spark.ml.util.TestingUtils._
 import org.apache.spark.sql.{Dataset, Row}
 
 
-class GaussianMixtureSuite extends MLTest
-  with DefaultReadWriteTest {
+class GaussianMixtureSuite extends MLTest with DefaultReadWriteTest {
 
   import testImplicits._
   import Encoders._

mllib/src/test/scala/org/apache/spark/ml/clustering/LDASuite.scala

@@ -23,8 +23,6 @@ import org.apache.spark.ml.linalg.{Vector, Vectors}
 import org.apache.spark.ml.util.{DefaultReadWriteTest, MLTest, MLTestingUtils}
 import org.apache.spark.ml.util.TestingUtils._
 import org.apache.spark.sql._
-import org.apache.spark.sql.catalyst.encoders.ExpressionEncoder
-
 
 object LDASuite {
   def generateLDAData(
@@ -62,8 +60,7 @@ object LDASuite {
 class LDASuite extends MLTest with DefaultReadWriteTest {
 
   import testImplicits._
-
-  implicit val vectorEncoder = ExpressionEncoder[Vector]()
+  import Encoders._
 
   val k: Int = 5
   val vocabSize: Int = 30