address viirya comments

xuanyuanking · xuanyuanking · commit 210d9420528e · 2018-11-14T20:46:45.000+08:00
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/clustering/KMeansSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/clustering/KMeansSuite.scala
@@ -19,8 +19,6 @@ package org.apache.spark.mllib.clustering
 
 import scala.util.Random
 
-import org.scalatest.Assertions
-
 import org.apache.spark.{SparkConf, SparkFunSuite}
 import org.apache.spark.mllib.linalg.{DenseVector, SparseVector, Vector, Vectors}
 import org.apache.spark.mllib.util.{LocalClusterSparkContext, MLlibTestSparkContext}
@@ -333,7 +331,7 @@ class KMeansSuite extends SparkFunSuite with MLlibTestSparkContext {
   }
 }
 
-object KMeansSuite extends SparkFunSuite with Assertions {
+object KMeansSuite extends SparkFunSuite {
   def createModel(dim: Int, k: Int, isSparse: Boolean): KMeansModel = {
     val singlePoint = if (isSparse) {
       Vectors.sparse(dim, Array.empty[Int], Array.empty[Double])

Original file line number	Diff line number	Diff line change
`@@ -19,8 +19,6 @@ package org.apache.spark.mllib.clustering`
`19`	`19`
`20`	`20`	`import scala.util.Random`
`21`	`21`
`22`		`-import org.scalatest.Assertions`
`23`		`-`
`24`	`22`	`import org.apache.spark.{SparkConf, SparkFunSuite}`
`25`	`23`	`import org.apache.spark.mllib.linalg.{DenseVector, SparseVector, Vector, Vectors}`
`26`	`24`	`import org.apache.spark.mllib.util.{LocalClusterSparkContext, MLlibTestSparkContext}`
`@@ -333,7 +331,7 @@ class KMeansSuite extends SparkFunSuite with MLlibTestSparkContext {`
`333`	`331`	`}`
`334`	`332`	`}`
`335`	`333`
`336`		`-object KMeansSuite extends SparkFunSuite with Assertions {`
	`334`	`+object KMeansSuite extends SparkFunSuite {`
`337`	`335`	`def createModel(dim: Int, k: Int, isSparse: Boolean): KMeansModel = {`
`338`	`336`	`val singlePoint = if (isSparse) {`
`339`	`337`	`Vectors.sparse(dim, Array.empty[Int], Array.empty[Double])`