summaryrefslogtreecommitdiffstats
path: root/tests/pool/fix_pool.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/pool/fix_pool.py')
-rw-r--r--tests/pool/fix_pool.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/pool/fix_pool.py b/tests/pool/fix_pool.py
new file mode 100644
index 0000000..12e4f39
--- /dev/null
+++ b/tests/pool/fix_pool.py
@@ -0,0 +1,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)