X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=test%2Fexecution%2Fscheduler.py;fp=test%2Fexecution%2Fscheduler.py;h=a7e3e9eecdb6b5bfccdf7a8ca9eb1898e15a89e2;hb=3fe2e6f7812888dc1366915545dd2243ff6fb1bb;hp=b36a86bd51e47e1b167cf2aabb8d5a0abeae6b9b;hpb=40abd97d4963775e6ec31b2b2ea80d2fd0642b7e;p=nepi.git

diff --git a/test/execution/scheduler.py b/test/execution/scheduler.py
index b36a86bd..a7e3e9ee 100755
--- a/test/execution/scheduler.py
+++ b/test/execution/scheduler.py
@@ -22,6 +22,8 @@ from nepi.util.timefuncs import tnow, stabsformat
 
 import unittest
 
+from six import next
+
 class SchedulerTestCase(unittest.TestCase):
     def test_task_order(self):
         def first():
@@ -49,13 +51,13 @@ class SchedulerTestCase(unittest.TestCase):
         scheduler.schedule(tsk1)
 
         # Make sure tasks are retrieved in teh correct order
-        tsk = scheduler.next()
+        tsk = next(scheduler)
         self.assertEqual(tsk.callback(), 1)
         
-        tsk = scheduler.next()
+        tsk = next(scheduler)
         self.assertEqual(tsk.callback(), 2)
         
-        tsk = scheduler.next()
+        tsk = next(scheduler)
         self.assertEqual(tsk.callback(), 3)