conftest.py

import signal
from contextlib import contextmanager
from typing import Generator

import pytest


def pytest_addoption(parser):
    parser.addoption(
        "--run-slow", action="store_true", default=False, help="run slow tests"
    )


def pytest_configure(config):
    config.addinivalue_line("markers", "slow: mark test as slow to run")


def pytest_collection_modifyitems(config, items):
    if config.getoption("--run-slow"):
        # --run-slow given in cli: do not skip slow tests
        return
    skip_slow = pytest.mark.skip(reason="need --run-slow option to run")
    for item in items:
        if "slow" in item.keywords:
            item.add_marker(skip_slow)


class TimeoutException(Exception):
    pass


@contextmanager
def time_limit(max_secs: int) -> Generator[None, None, None]:
    def signal_handler(signum: int, frame):
        raise TimeoutException('Timed out!')
    signal.signal(signal.SIGALRM, signal_handler)
    signal.alarm(max_secs)
    try:
        yield
    finally:
        signal.alarm(0)