Skip to content

Commit 50a2fd6

Browse files
committed
Fix test suite issue after moving aggregator to Shuffle reader and writer
1 parent 1a96190 commit 50a2fd6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

core/src/test/scala/org/apache/spark/scheduler/SparkListenerSuite.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ class SparkListenerSuite extends FunSuite with LocalSparkContext with Matchers
181181
assert(sc.listenerBus.waitUntilEmpty(WAIT_TIMEOUT_MILLIS))
182182
listener.stageInfos.size should be {2} // Shuffle map stage + result stage
183183
val stageInfo3 = listener.stageInfos.keys.find(_.stageId == 2).get
184-
stageInfo3.rddInfos.size should be {2} // ShuffledRDD, MapPartitionsRDD
184+
stageInfo3.rddInfos.size should be {1} // ShuffledRDD
185185
stageInfo3.rddInfos.forall(_.numPartitions == 4) should be {true}
186186
stageInfo3.rddInfos.exists(_.name == "Trois") should be {true}
187187
}

0 commit comments

Comments
 (0)