Code review

2 files changed, 3 insertions, 3 deletions

diff --git a/core/src/main/scala/spark/rdd/PartitionPruningRDD.scala b/core/src/main/scala/spark/rdd/PartitionPruningRDD.scala
index eaac134ef4..fa2f5c2611 100644
--- a/core/src/main/scala/spark/rdd/PartitionPruningRDD.scala
+++ b/core/src/main/scala/spark/rdd/PartitionPruningRDD.scala
@@ -17,8 +17,8 @@ class PruneDependency[T](rdd: RDD[T], @transient partitionFilterFunc: Int => Boo
 
   @transient
   val partitions: Array[Partition] = rdd.partitions.zipWithIndex
-  .filter(s => partitionFilterFunc(s._2))
-  .map { case(split, idx) => new PartitionPruningRDDPartition(idx, split) : Partition }
+    .filter(s => partitionFilterFunc(s._2))
+    .map { case(split, idx) => new PartitionPruningRDDPartition(idx, split) : Partition }
 
   override def getParents(partitionId: Int) = List(partitions(partitionId).index)
 }
diff --git a/core/src/test/scala/spark/PartitionPruningRDDSuite.scala b/core/src/test/scala/spark/PartitionPruningRDDSuite.scala
index 4fd1f55a45..88352b639f 100644
--- a/core/src/test/scala/spark/PartitionPruningRDDSuite.scala
+++ b/core/src/test/scala/spark/PartitionPruningRDDSuite.scala
@@ -21,8 +21,8 @@ class PartitionPruningRDDSuite extends FunSuite with SharedSparkContext {
       def compute(split: Partition, context: TaskContext) = {Iterator()}
     }
     val prunedRDD = PartitionPruningRDD.create(rdd, {x => if (x==2) true else false})
-    println(prunedRDD.partitions.length)
     val p = prunedRDD.partitions(0)
     assert(p.index == 2)
+    assert(prunedRDD.partitions.length == 1)
   }
 }