@@ -19,25 +19,15 @@ package org.apache.spark.scheduler
1919
2020import org .apache .spark .{SparkConf , SparkException , SparkContext }
2121import org .apache .spark .util .{SerializableBuffer , AkkaUtils }
22- import org .apache .spark .SparkContext ._
2322
24- import org .scalatest .{ BeforeAndAfterAll , BeforeAndAfter , FunSuite }
23+ import org .scalatest .FunSuite
2524
26- class CoarseGrainedSchedulerBackendSuite extends FunSuite with
27- BeforeAndAfter with BeforeAndAfterAll {
25+ class CoarseGrainedSchedulerBackendSuite extends FunSuite {
2826
29- override def beforeAll {
30- System .setProperty(" spark.akka.frameSize" , " 1" )
31- System .setProperty(" spark.default.parallelism" , " 1" )
32- }
33-
34- override def afterAll {
35- System .clearProperty(" spark.akka.frameSize" )
36- System .clearProperty(" spark.default.parallelism" )
37- }
38-
39- test(" serialized task larger than Akka frame size" ) {
27+ test(" serialized task larger than akka frame size" ) {
4028 val conf = new SparkConf
29+ conf.set(" spark.akka.frameSize" ," 1" )
30+ conf.set(" spark.default.parallelism" ," 1" )
4131 val sc = new SparkContext (" local-cluster[2 , 1 , 512]" , " test" , conf)
4232 val frameSize = AkkaUtils .maxFrameSizeBytes(sc.conf)
4333 val buffer = new SerializableBuffer (java.nio.ByteBuffer .allocate(2 * frameSize))
0 commit comments