36 lines
994 B
Python
Executable File
36 lines
994 B
Python
Executable File
#!/usr/bin/env python
|
|
'''
|
|
testit.py - Example code for ad-hoc function timeouts.
|
|
|
|
Newer tests for all features found in "tests" directory.
|
|
'''
|
|
|
|
from func_timeout import func_timeout, FunctionTimedOut
|
|
import time
|
|
import sys
|
|
|
|
|
|
def doit(howmany):
|
|
time.sleep(2)
|
|
return 17 + howmany
|
|
|
|
if __name__ == '__main__':
|
|
|
|
print ( "Should get return value of 23:" )
|
|
print ( "\tGot Return: %s\n" %(str(func_timeout(4, doit, args=(6,))),) )
|
|
|
|
print ( "\nShould time out (exception):" )
|
|
myException = None
|
|
try:
|
|
print ("\tGot Return: %s\n" %(str(func_timeout(1, doit, kwargs={'howmany' : 16})),))
|
|
except FunctionTimedOut as e:
|
|
sys.stderr.write('\tGot Exception: %s\n' %(str(e),))
|
|
myException = e
|
|
pass
|
|
|
|
print ( "\nRetrying with longer timeout, should get 16+17=33:" )
|
|
if myException is not None:
|
|
print ( "\nGot: %s\n" %( str(myException.retry(2.5)), ) )
|
|
else:
|
|
sys.stderr.write('Did not get exception before?\n')
|