diff options
author | Eric Liang <ekl@google.com> | 2014-09-07 17:57:59 -0700 |
---|---|---|
committer | Matei Zaharia <matei@databricks.com> | 2014-09-07 17:57:59 -0700 |
commit | 6754570d83044c4fbaf0d2ac2378a0e081a93629 (patch) | |
tree | d42b454b67cccbebcdc7e04186d924d4906f9b04 /core/src/test | |
parent | 3fb57a0ab3d76fda2301dbe9f2f3fa6743b4ed78 (diff) | |
download | spark-6754570d83044c4fbaf0d2ac2378a0e081a93629.tar.gz spark-6754570d83044c4fbaf0d2ac2378a0e081a93629.tar.bz2 spark-6754570d83044c4fbaf0d2ac2378a0e081a93629.zip |
[SPARK-3394] [SQL] Fix crash in TakeOrdered when limit is 0
This resolves https://issues.apache.org/jira/browse/SPARK-3394
Author: Eric Liang <ekl@google.com>
Closes #2264 from ericl/spark-3394 and squashes the following commits:
c87355b [Eric Liang] refactor
bfb6140 [Eric Liang] change RDD takeOrdered instead
7a51528 [Eric Liang] fix takeordered when limit = 0
Diffstat (limited to 'core/src/test')
-rw-r--r-- | core/src/test/scala/org/apache/spark/rdd/RDDSuite.scala | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/core/src/test/scala/org/apache/spark/rdd/RDDSuite.scala b/core/src/test/scala/org/apache/spark/rdd/RDDSuite.scala index 926d4fecb5..499dcda3da 100644 --- a/core/src/test/scala/org/apache/spark/rdd/RDDSuite.scala +++ b/core/src/test/scala/org/apache/spark/rdd/RDDSuite.scala @@ -521,6 +521,13 @@ class RDDSuite extends FunSuite with SharedSparkContext { assert(sortedLowerK === Array(1, 2, 3, 4, 5)) } + test("takeOrdered with limit 0") { + val nums = Array(1, 2, 3, 4, 5, 6, 7, 8, 9, 10) + val rdd = sc.makeRDD(nums, 2) + val sortedLowerK = rdd.takeOrdered(0) + assert(sortedLowerK.size === 0) + } + test("takeOrdered with custom ordering") { val nums = Array(1, 2, 3, 4, 5, 6, 7, 8, 9, 10) implicit val ord = implicitly[Ordering[Int]].reverse |