diff options
author | Giampaolo RodolĂ <g.rodola@gmail.com> | 2010-08-04 09:28:05 (GMT) |
---|---|---|
committer | Giampaolo RodolĂ <g.rodola@gmail.com> | 2010-08-04 09:28:05 (GMT) |
commit | b5c23761d3ef63d596d5f0c1f1e510b30fdf21e1 (patch) | |
tree | 155bdf9de9452e6c9ff84841f4f3f0d1ae9e816d /Lib/test/test_sched.py | |
parent | 934abddaece303fce61e1ab3bb3c631c30f117c2 (diff) | |
download | cpython-b5c23761d3ef63d596d5f0c1f1e510b30fdf21e1.zip cpython-b5c23761d3ef63d596d5f0c1f1e510b30fdf21e1.tar.gz cpython-b5c23761d3ef63d596d5f0c1f1e510b30fdf21e1.tar.bz2 |
issue #8687: provides a test suite for sched.py module
Diffstat (limited to 'Lib/test/test_sched.py')
-rw-r--r-- | Lib/test/test_sched.py | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/Lib/test/test_sched.py b/Lib/test/test_sched.py new file mode 100644 index 0000000..29fd277 --- /dev/null +++ b/Lib/test/test_sched.py @@ -0,0 +1,80 @@ +#!/usr/bin/env python + +import sched +import time +import unittest +from test import support + + +class TestCase(unittest.TestCase): + + def test_enter(self): + l = [] + fun = lambda x: l.append(x) + scheduler = sched.scheduler(time.time, time.sleep) + for x in [0.05, 0.04, 0.03, 0.02, 0.01]: + z = scheduler.enter(x, 1, fun, (x,)) + scheduler.run() + self.assertEqual(l, [0.01, 0.02, 0.03, 0.04, 0.05]) + + def test_enterabs(self): + l = [] + fun = lambda x: l.append(x) + scheduler = sched.scheduler(time.time, time.sleep) + for x in [0.05, 0.04, 0.03, 0.02, 0.01]: + z = scheduler.enterabs(x, 1, fun, (x,)) + scheduler.run() + self.assertEqual(l, [0.01, 0.02, 0.03, 0.04, 0.05]) + + def test_priority(self): + l = [] + fun = lambda x: l.append(x) + scheduler = sched.scheduler(time.time, time.sleep) + for priority in [1, 2, 3, 4, 5]: + z = scheduler.enter(0.01, priority, fun, (priority,)) + scheduler.run() + self.assertEqual(l, [1, 2, 3, 4, 5]) + + def test_cancel(self): + l = [] + fun = lambda x: l.append(x) + scheduler = sched.scheduler(time.time, time.sleep) + event1 = scheduler.enter(0.01, 1, fun, (0.01,)) + event2 = scheduler.enter(0.02, 1, fun, (0.02,)) + event3 = scheduler.enter(0.03, 1, fun, (0.03,)) + event4 = scheduler.enter(0.04, 1, fun, (0.04,)) + event5 = scheduler.enter(0.05, 1, fun, (0.05,)) + scheduler.cancel(event1) + scheduler.cancel(event5) + scheduler.run() + self.assertEqual(l, [0.02, 0.03, 0.04]) + + def test_empty(self): + l = [] + fun = lambda x: l.append(x) + scheduler = sched.scheduler(time.time, time.sleep) + self.assertTrue(scheduler.empty()) + for x in [0.05, 0.04, 0.03, 0.02, 0.01]: + z = scheduler.enterabs(x, 1, fun, (x,)) + self.assertFalse(scheduler.empty()) + scheduler.run() + self.assertTrue(scheduler.empty()) + + def test_queue(self): + l = [] + events = [] + fun = lambda x: l.append(x) + scheduler = sched.scheduler(time.time, time.sleep) + self.assertEqual(scheduler._queue, []) + for x in [0.05, 0.04, 0.03, 0.02, 0.01]: + events.append(scheduler.enterabs(x, 1, fun, (x,))) + self.assertEqual(scheduler._queue.sort(), events.sort()) + scheduler.run() + self.assertEqual(scheduler._queue, []) + + +def test_main(): + support.run_unittest(TestCase) + +if __name__ == "__main__": + test_main() |