Use pytest instead of unittest

Author: Cito

setup.py

@@ -61,6 +61,7 @@
     license='MIT License',
     packages=['dbutils'],
     extras_require={
-        "pg": ["PyGreSQL >= 5"]
-    }
+        "pg": ["PyGreSQL>=5"]
+    },
+    tests_require=["pytest>=7", "ruff"]
 )

tests/__init__.py

@@ -1 +1,4 @@
-# DBUtils tests
+"""DBUtils tests"""
+
+# make sure the mock pg module is installed
+from . import mock_pg as pg  # noqa: F401

tests/mock_db.py

@@ -1,8 +1,23 @@
 """This module serves as a mock object for the DB-API 2 module"""
 
+import sys
+
+import pytest
+
+__all__ = ['dbapi']
+
+
 threadsafety = 2
 
 
+@pytest.fixture()
+def dbapi():
+    """Get mock DB API 2 module."""
+    mock_db = sys.modules[__name__]
+    mock_db.threadsafety = 2
+    return mock_db
+
+
 class Error(Exception):
     pass
 

tests/mock_pg.py

@@ -2,6 +2,7 @@
 
 import sys
 
+
 sys.modules['pg'] = sys.modules[__name__]