23 lines
446 B
Python
23 lines
446 B
Python
|
#! /usr/bin/env python3
|
||
|
|
# vim:fenc=utf-8
|
||
|
|
|
||
|
|
import time
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from src.time_out import TimeoutException, time_limit
|
||
|
|
|
||
|
|
|
||
|
|
class TestTimeOut(unittest.TestCase):
|
||
|
|
def test_exception(self):
|
||
|
|
with self.assertRaises(TimeoutException):
|
||
|
|
with time_limit(1):
|
||
|
|
time.sleep(2)
|
||
|
|
|
||
|
|
def test_no_exception(self):
|
||
|
|
with time_limit(1):
|
||
|
|
time.sleep(0.5)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|