diff options
Diffstat (limited to 'core')
-rw-r--r-- | core/src/test/scala/org/apache/spark/api/python/PythonRDDSuite.scala | 1 | ||||
-rw-r--r-- | core/src/test/scala/org/apache/spark/rdd/RDDSuite.scala | 2 |
2 files changed, 1 insertions, 2 deletions
diff --git a/core/src/test/scala/org/apache/spark/api/python/PythonRDDSuite.scala b/core/src/test/scala/org/apache/spark/api/python/PythonRDDSuite.scala index 1bebfe5ec8..5bcebabc9a 100644 --- a/core/src/test/scala/org/apache/spark/api/python/PythonRDDSuite.scala +++ b/core/src/test/scala/org/apache/spark/api/python/PythonRDDSuite.scala @@ -19,7 +19,6 @@ package org.apache.spark.api.python import org.scalatest.FunSuite import org.scalatest.matchers.ShouldMatchers -import org.apache.spark.api.python.PythonRDD import java.io.{ByteArrayOutputStream, DataOutputStream} 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 cd01303bad..223ebec5fa 100644 --- a/core/src/test/scala/org/apache/spark/rdd/RDDSuite.scala +++ b/core/src/test/scala/org/apache/spark/rdd/RDDSuite.scala @@ -48,7 +48,7 @@ class RDDSuite extends FunSuite with SharedSparkContext { val partitionSums = nums.mapPartitions(iter => Iterator(iter.reduceLeft(_ + _))) assert(partitionSums.collect().toList === List(3, 7)) - val partitionSumsWithSplit = nums.mapPartitionsWithSplit { + val partitionSumsWithSplit = nums.mapPartitionsWithIndex { case(split, iter) => Iterator((split, iter.reduceLeft(_ + _))) } assert(partitionSumsWithSplit.collect().toList === List((0, 3), (1, 7))) |