fix failed test cases

apache · Aug 4, 2014 · 2ba6195 · 2ba6195
1 parent 87dd09e
commit 2ba6195
Showing 1 changed file with 2 additions and 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
@@ -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