36 lines
975 B
Python
Executable File
36 lines
975 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 ( f"\tGot Return: {str(func_timeout(4, doit, args=(6,)))}\n" )
|
|
|
|
print ( "\nShould time out (exception):" )
|
|
myException = None
|
|
try:
|
|
print (f"\tGot Return: {str(func_timeout(1, doit, kwargs={'howmany': 16}))}\n")
|
|
except FunctionTimedOut as e:
|
|
sys.stderr.write(f'\tGot Exception: {str(e)}\n')
|
|
myException = e
|
|
pass
|
|
|
|
print ( "\nRetrying with longer timeout, should get 16+17=33:" )
|
|
if myException is not None:
|
|
print ( f"\nGot: {str(myException.retry(2.5))}\n" )
|
|
else:
|
|
sys.stderr.write('Did not get exception before?\n')
|