From 733d6ddf51ddf440efb1a17c818da6d7fd027c4b Mon Sep 17 00:00:00 2001 From: Manish Amde Date: Sun, 22 Dec 2013 12:20:50 -0800 Subject: [PATCH] fixed tests Signed-off-by: Manish Amde --- .../apache/spark/mllib/tree/DecisionTreeSuite.scala | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/mllib/src/test/scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala index 2b5988bb3c6a3..6b6bb2b55e7b5 100644 --- a/mllib/src/test/scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala @@ -53,7 +53,7 @@ class DecisionTreeSuite extends FunSuite with BeforeAndAfterAll { assert(bins.length==2) assert(splits(0).length==99) assert(bins(0).length==100) - println(splits(1)(98)) + //println(splits(1)(98)) } test("stump with fixed label 0 for Gini"){ @@ -68,7 +68,10 @@ class DecisionTreeSuite extends FunSuite with BeforeAndAfterAll { assert(bins(0).length==100) assert(splits(0).length==99) assert(bins(0).length==100) - val bestSplits = DecisionTree.findBestSplits(rdd,Array(0.0),strategy,0,Array[List[Filter]](),splits,bins) + + strategy.numBins = 100 + val bestSplits = DecisionTree.findBestSplits(rdd,new Array(7),strategy,0,Array[List[Filter]](),splits,bins) + println("here") assert(bestSplits.length == 1) assert(0==bestSplits(0)._1.feature) assert(10==bestSplits(0)._1.threshold) @@ -91,6 +94,8 @@ class DecisionTreeSuite extends FunSuite with BeforeAndAfterAll { assert(bins(0).length==100) assert(splits(0).length==99) assert(bins(0).length==100) + + strategy.numBins = 100 val bestSplits = DecisionTree.findBestSplits(rdd,Array(0.0),strategy,0,Array[List[Filter]](),splits,bins) assert(bestSplits.length == 1) assert(0==bestSplits(0)._1.feature) @@ -115,6 +120,8 @@ class DecisionTreeSuite extends FunSuite with BeforeAndAfterAll { assert(bins(0).length==100) assert(splits(0).length==99) assert(bins(0).length==100) + + strategy.numBins = 100 val bestSplits = DecisionTree.findBestSplits(rdd,Array(0.0),strategy,0,Array[List[Filter]](),splits,bins) assert(bestSplits.length == 1) assert(0==bestSplits(0)._1.feature) @@ -138,6 +145,8 @@ class DecisionTreeSuite extends FunSuite with BeforeAndAfterAll { assert(bins(0).length==100) assert(splits(0).length==99) assert(bins(0).length==100) + + strategy.numBins = 100 val bestSplits = DecisionTree.findBestSplits(rdd,Array(0.0),strategy,0,Array[List[Filter]](),splits,bins) assert(bestSplits.length == 1) assert(0==bestSplits(0)._1.feature)