You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
678 B
25 lines
678 B
5 years ago
|
import time
|
||
|
|
from multiprocessing import Process
|
||
|
|
|
||
|
|
# Note: this does not return any return values of the function, just the exit status
|
||
|
|
INTERVAL = 0.1
|
||
|
|
def run_with_timeout(timeout, fn, *kwargs):
|
||
|
|
def runner(fn, kwargs):
|
||
|
|
try:
|
||
|
|
fn(*kwargs)
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|
||
|
|
raise e
|
||
|
|
|
||
|
|
process = Process(target=runner, args=(fn, kwargs))
|
||
|
|
process.start()
|
||
|
|
|
||
|
|
counter = 0
|
||
|
|
while process.is_alive():
|
||
|
|
time.sleep(INTERVAL)
|
||
|
|
counter+=1
|
||
|
|
if (counter * INTERVAL) > timeout:
|
||
|
|
process.terminate()
|
||
|
|
raise TimeoutError("Function timed out!")
|
||
|
|
if process.exitcode != 0:
|
||
|
|
raise RuntimeError("Test failed with exit code: ", str(process.exitcode))
|