Permalink
Browse files

Added a unit test for PairRDDFunctions.lookup

Lookup didn't have a unit test. Added two tests, one for with a partitioner, and one for without.

Author: Bryn Keller <bryn.keller@intel.com>

Closes #36 from xoltar/lookup and squashes the following commits:

3bc0d44 [Bryn Keller] Added a unit test for PairRDDFunctions.lookup
  • Loading branch information...
1 parent b55cade commit 923dba5096d4f7a96d67e4ee243b3b1085984bb9 @xoltar xoltar committed with mateiz Mar 4, 2014
Showing with 26 additions and 0 deletions.
  1. +26 −0 core/src/test/scala/org/apache/spark/rdd/PairRDDFunctionsSuite.scala
@@ -347,6 +347,32 @@ class PairRDDFunctionsSuite extends FunSuite with SharedSparkContext {
*/
pairs.saveAsNewAPIHadoopFile[ConfigTestFormat]("ignored")
}
+
+ test("lookup") {
+ val pairs = sc.parallelize(Array((1,2), (3,4), (5,6), (5,7)))
+
+ assert(pairs.partitioner === None)
+ assert(pairs.lookup(1) === Seq(2))
+ assert(pairs.lookup(5) === Seq(6,7))
+ assert(pairs.lookup(-1) === Seq())
+
+ }
+
+ test("lookup with partitioner") {
+ val pairs = sc.parallelize(Array((1,2), (3,4), (5,6), (5,7)))
+
+ val p = new Partitioner {
+ def numPartitions: Int = 2
+
+ def getPartition(key: Any): Int = Math.abs(key.hashCode() % 2)
+ }
+ val shuffled = pairs.partitionBy(p)
+
+ assert(shuffled.partitioner === Some(p))
+ assert(shuffled.lookup(1) === Seq(2))
+ assert(shuffled.lookup(5) === Seq(6,7))
+ assert(shuffled.lookup(-1) === Seq())
+ }
}
/*

0 comments on commit 923dba5

Please sign in to comment.