File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed
mllib/src/test/scala/org/apache/spark/ml/feature Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change 1717
1818package org .apache .spark .ml .feature
1919
20- import org .scalatest .FunSuite
21-
22- import org .apache .spark .SparkContext
2320import org .apache .spark .ml .attribute .{Attribute , NominalAttribute }
2421import org .apache .spark .mllib .util .MLlibTestSparkContext
2522import org .apache .spark .sql .{Row , SQLContext }
23+ import org .apache .spark .{SparkContext , SparkFunSuite }
2624
27- class QuantileDiscretizerSuite extends FunSuite with MLlibTestSparkContext {
25+ class QuantileDiscretizerSuite extends SparkFunSuite with MLlibTestSparkContext {
2826 import org .apache .spark .ml .feature .QuantileDiscretizerSuite ._
2927
3028 test(" Test quantile discretizer" ) {
@@ -55,7 +53,7 @@ class QuantileDiscretizerSuite extends FunSuite with MLlibTestSparkContext {
5553 }
5654}
5755
58- private object QuantileDiscretizerSuite extends FunSuite {
56+ private object QuantileDiscretizerSuite extends SparkFunSuite {
5957
6058 def checkDiscretizedData (
6159 sc : SparkContext ,
You can’t perform that action at this time.
0 commit comments