summaryrefslogtreecommitdiffstats
path: root/tests/pool/fix_pool.py
blob: 12e4f3941bfec22928fa997ecb768fcdf3bab50b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
import pytest


def pytest_configure(config):
    config.addinivalue_line("markers", "pool: test related to the psycopg_pool package")


def pytest_collection_modifyitems(items):
    # Add the pool markers to all the tests in the pool package
    for item in items:
        if "/pool/" in item.nodeid:
            item.add_marker(pytest.mark.pool)