93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
#!/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_args_kwargs(self):
|
|
flag = []
|
|
|
|
def fun(*a, **b):
|
|
flag.append(None)
|
|
self.assertEqual(a, (1,2,3))
|
|
self.assertEqual(b, {"foo":1})
|
|
|
|
scheduler = sched.scheduler(time.time, time.sleep)
|
|
z = scheduler.enterabs(0.01, 1, fun, argument=(1,2,3), kwargs={"foo":1})
|
|
scheduler.run()
|
|
self.assertEqual(flag, [None])
|
|
|
|
def test_main():
|
|
support.run_unittest(TestCase)
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|