Fail a unittest after X seconds

unittest_timeout.py

import signal
import unittest
from time import sleep
 
class TimeoutException(Exception):
    pass
 
 
def signal_handler(signum, frame):
    raise TimeoutException
 
 
class FailAfter(object):
    def __init__(self, seconds):
        self.seconds = seconds
 
    def __call__(self, f):
        def timed_f(*args, **kwargs):
            res = None
            signal.signal(signal.SIGALRM, signal_handler)
            signal.alarm(self.seconds)
            try:
                res = f(*args, **kwargs)
            except TimeoutException:
                args[0].fail()
            signal.alarm(0)
            return res
        return timed_f
        
 
class Test(unittest.TestCase):
    @FailAfter(seconds=10)
    def test_something(self):
        while True:
            sleep(1)
            print("bla")
    
    @FailAfter(2)
    def test_something_fast(self):
        self.assertTrue(1==1)