From 2ba61950f666c033a11b889e02dcfa3b1201ccb5 Mon Sep 17 00:00:00 2001 From: CodingCat Date: Thu, 31 Jul 2014 21:38:19 -0400 Subject: [PATCH] fix failed test cases --- .../org/apache/spark/scheduler/TaskSetManagerSuite.scala | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/core/src/test/scala/org/apache/spark/scheduler/TaskSetManagerSuite.scala b/core/src/test/scala/org/apache/spark/scheduler/TaskSetManagerSuite.scala index 076afa26a42ed..79d503e3d9374 100644 --- a/core/src/test/scala/org/apache/spark/scheduler/TaskSetManagerSuite.scala +++ b/core/src/test/scala/org/apache/spark/scheduler/TaskSetManagerSuite.scala @@ -573,10 +573,8 @@ class TaskSetManagerSuite extends FunSuite with LocalSparkContext with Logging { assert(manager.resourceOffer("execA", "host1", PROCESS_LOCAL).get.index === 0) assert(manager.resourceOffer("execA", "host1", NODE_LOCAL) == None) - assert(manager.resourceOffer("execA", "host1", NO_PREF) == None) - clock.advance(LOCALITY_WAIT) - // schedule a node local task - assert(manager.resourceOffer("execA", "host1", NODE_LOCAL).get.index === 1) + assert(manager.resourceOffer("execA", "host1", NO_PREF).get.index == 1) + manager.speculatableTasks += 1 clock.advance(LOCALITY_WAIT) // schedule the nonPref task