From c24b3819dd474e13d6098150c174b2e7e4bc6498 Mon Sep 17 00:00:00 2001
From: Reynold Xin <rxin@cs.berkeley.edu>
Date: Wed, 23 Jan 2013 15:34:59 -0800
Subject: [PATCH] Added an extra assert for split size check.
---
core/src/test/scala/spark/RDDSuite.scala | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/core/src/test/scala/spark/RDDSuite.scala b/core/src/test/scala/spark/RDDSuite.scala
index 03aa2845f4..ef74c99246 100644
--- a/core/src/test/scala/spark/RDDSuite.scala
+++ b/core/src/test/scala/spark/RDDSuite.scala
@@ -173,7 +173,10 @@ class RDDSuite extends FunSuite with BeforeAndAfter {
sc = new SparkContext("local", "test")
val data = sc.parallelize(1 to 10, 10)
// Note that split number starts from 0, so > 8 means only 10th partition left.
- val prunedData = data.pruneSplits(splitNum => splitNum > 8).collect
- assert(prunedData.size == 1 && prunedData(0) == 10)
+ val prunedRdd = data.pruneSplits(splitNum => splitNum > 8)
+ assert(prunedRdd.splits.size == 1)
+ val prunedData = prunedRdd.collect
+ assert(prunedData.size == 1)
+ assert(prunedData(0) == 10)
}
}
--
GitLab