Skip to content

Commit 66a53a6

Browse files
committed
[HOTFIX] Replace FunSuite with SparkFunSuite.
This fixes a build break introduced by merging a643002, which fails the new style checks that ensure that we use SparkFunSuite instead of FunSuite.
1 parent 1281a35 commit 66a53a6

File tree

3 files changed

+4
-9
lines changed

3 files changed

+4
-9
lines changed

core/src/test/scala/org/apache/spark/shuffle/sort/BypassMergeSortShuffleWriterSuite.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,15 +29,15 @@ import org.mockito.Matchers._
2929
import org.mockito.Mockito._
3030
import org.mockito.invocation.InvocationOnMock
3131
import org.mockito.stubbing.Answer
32-
import org.scalatest.{BeforeAndAfterEach, FunSuite}
32+
import org.scalatest.BeforeAndAfterEach
3333

3434
import org.apache.spark._
3535
import org.apache.spark.executor.{TaskMetrics, ShuffleWriteMetrics}
3636
import org.apache.spark.serializer.{SerializerInstance, Serializer, JavaSerializer}
3737
import org.apache.spark.storage._
3838
import org.apache.spark.util.Utils
3939

40-
class BypassMergeSortShuffleWriterSuite extends FunSuite with BeforeAndAfterEach {
40+
class BypassMergeSortShuffleWriterSuite extends SparkFunSuite with BeforeAndAfterEach {
4141

4242
@Mock(answer = RETURNS_SMART_NULLS) private var blockManager: BlockManager = _
4343
@Mock(answer = RETURNS_SMART_NULLS) private var diskBlockManager: DiskBlockManager = _

core/src/test/scala/org/apache/spark/shuffle/sort/SortShuffleWriterSuite.scala

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,10 @@
1818
package org.apache.spark.shuffle.sort
1919

2020
import org.mockito.Mockito._
21-
import org.scalatest.FunSuite
2221

23-
import org.apache.spark.{Aggregator, SparkConf}
22+
import org.apache.spark.{Aggregator, SparkConf, SparkFunSuite}
2423

25-
class SortShuffleWriterSuite extends FunSuite {
24+
class SortShuffleWriterSuite extends SparkFunSuite {
2625

2726
import SortShuffleWriter._
2827

core/src/test/scala/org/apache/spark/util/collection/ExternalSorterSuite.scala

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,8 @@ package org.apache.spark.util.collection
1919

2020
import scala.collection.mutable.ArrayBuffer
2121

22-
import org.scalatest.PrivateMethodTester
23-
2422
import scala.util.Random
2523

26-
import org.scalatest.FunSuite
27-
2824
import org.apache.spark._
2925
import org.apache.spark.serializer.{JavaSerializer, KryoSerializer}
3026

0 commit comments

Comments
 (0)