TST: keep compact and open testParams

apache · Mar 28, 2017 · 5e0fbe3 · 5e0fbe3
1 parent 55fe603
commit 5e0fbe3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/mllib/src/test/scala/org/apache/spark/ml/classification/DecisionTreeClassifierSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/classification/DecisionTreeClassifierSuite.scala
@@ -389,17 +389,15 @@ class DecisionTreeClassifierSuite
   test("SPARK-20043: " +
        "ImpurityCalculator builder fails for uppercase impurity type Gini in model read/write") {
     val rdd = TreeTests.getTreeReadWriteData(sc)
-
     val data: DataFrame =
       TreeTests.setMetadata(rdd, Map.empty[Int, Int], numClasses = 2)
 
     val dt = new DecisionTreeClassifier()
       .setImpurity("Gini")
       .setMaxDepth(2)
-
     val model = dt.fit(data)
 
-    testDefaultReadWrite(model, false)
+    testDefaultReadWrite(model)
   }
 }
 

diff --git a/mllib/src/test/scala/org/apache/spark/ml/regression/DecisionTreeRegressorSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/regression/DecisionTreeRegressorSuite.scala
@@ -182,17 +182,15 @@ class DecisionTreeRegressorSuite
   test("SPARK-20043: " +
        "ImpurityCalculator builder fails for uppercase impurity type in model read/write") {
     val rdd = TreeTests.getTreeReadWriteData(sc)
-
     val data: DataFrame =
       TreeTests.setMetadata(rdd, Map.empty[Int, Int], numClasses = 0)
 
     val dt = new DecisionTreeRegressor()
       .setImpurity("Variance")
       .setMaxDepth(2)
-
     val model = dt.fit(data)
 
-    testDefaultReadWrite(model, false)
+    testDefaultReadWrite(model)
   }
 }