diff options
author | Xiangrui Meng <meng@databricks.com> | 2015-11-13 13:19:04 -0800 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-11-13 13:19:04 -0800 |
commit | bdfbc1dcaf121a1a1239857adcf54cdfe82c26dc (patch) | |
tree | 46f8ff272c38953ec4c23372b9bbd285fe400bbf | |
parent | 912b94363bb113ab14024a45f17a4d2b82a09e66 (diff) | |
download | spark-bdfbc1dcaf121a1a1239857adcf54cdfe82c26dc.tar.gz spark-bdfbc1dcaf121a1a1239857adcf54cdfe82c26dc.tar.bz2 spark-bdfbc1dcaf121a1a1239857adcf54cdfe82c26dc.zip |
[MINOR][ML] remove MLlibTestsSparkContext from ImpuritySuite
ImpuritySuite doesn't need SparkContext.
Author: Xiangrui Meng <meng@databricks.com>
Closes #9698 from mengxr/remove-mllib-test-context-in-impurity-suite.
-rw-r--r-- | mllib/src/test/scala/org/apache/spark/mllib/tree/ImpuritySuite.scala | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/tree/ImpuritySuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/tree/ImpuritySuite.scala index 49aff21fe7..14152cdd63 100644 --- a/mllib/src/test/scala/org/apache/spark/mllib/tree/ImpuritySuite.scala +++ b/mllib/src/test/scala/org/apache/spark/mllib/tree/ImpuritySuite.scala @@ -19,12 +19,11 @@ package org.apache.spark.mllib.tree import org.apache.spark.SparkFunSuite import org.apache.spark.mllib.tree.impurity.{EntropyAggregator, GiniAggregator} -import org.apache.spark.mllib.util.MLlibTestSparkContext /** * Test suites for [[GiniAggregator]] and [[EntropyAggregator]]. */ -class ImpuritySuite extends SparkFunSuite with MLlibTestSparkContext { +class ImpuritySuite extends SparkFunSuite { test("Gini impurity does not support negative labels") { val gini = new GiniAggregator(2) intercept[IllegalArgumentException] { |