fix test

Author: attilapiros

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

@@ -3378,7 +3378,7 @@ class DAGSchedulerSuite extends SparkFunSuite with TempLocalSparkContext with Ti
     // As the ShuffleMapStage is inDeterminate all the partitions need to be retried
     assert(failedStages.collect {
       case stage: ShuffleMapStage if stage.shuffleDep.shuffleId == shuffleId2 => stage
-    }.head.findMissingPartitions() == Seq(0, 1))
+    }.head.findMissingPartitions() == Seq(0))
     // The result stage is still waiting for its 2 tasks to complete
     assert(failedStages.collect {
       case stage: ResultStage => stage
@@ -4350,7 +4350,7 @@ class DAGSchedulerSuite extends SparkFunSuite with TempLocalSparkContext with Ti
     // As the ShuffleMapStage is inDeterminate all the partitions need to be retried
     assert(failedStages.collect {
       case stage: ShuffleMapStage if stage.shuffleDep.shuffleId == shuffleId2 => stage
-    }.head.findMissingPartitions() == Seq(0, 1))
+    }.head.findMissingPartitions() == Seq(0))
     // The result stage is still waiting for its 2 tasks to complete
     assert(failedStages.collect {
       case stage: ResultStage => stage