import unittest
+from six import next
+
class SchedulerTestCase(unittest.TestCase):
def test_task_order(self):
def first():
scheduler.schedule(tsk1)
# Make sure tasks are retrieved in teh correct order
- tsk = scheduler.next()
- self.assertEquals(tsk.callback(), 1)
+ tsk = next(scheduler)
+ self.assertEqual(tsk.callback(), 1)
- tsk = scheduler.next()
- self.assertEquals(tsk.callback(), 2)
+ tsk = next(scheduler)
+ self.assertEqual(tsk.callback(), 2)
- tsk = scheduler.next()
- self.assertEquals(tsk.callback(), 3)
+ tsk = next(scheduler)
+ self.assertEqual(tsk.callback(), 3)
if __name__ == '__main__':