Add type annotations to two-phase commit tests

diff --git a/tests/dbapi20_tpc.py b/tests/dbapi20_tpc.py
index d68703282b68a2bb13a10b6d87591361bf0f035a..5026a61249f84ba49c5698c874739081ec7ed7b0 100644 (file)
--- a/tests/dbapi20_tpc.py
+++ b/tests/dbapi20_tpc.py
@@ -6,11 +6,12 @@
 """
 
 import unittest
+from typing import Any
 
 
 class TwoPhaseCommitTests(unittest.TestCase):
 
-    driver = None
+    driver: Any = None
 
     def connect(self):
         """Make a database connection."""

diff --git a/tests/test_psycopg_dbapi20.py b/tests/test_psycopg_dbapi20.py
index 4aa1cc7c7d3bd8a2f84282323acb0b71a970750f..9c900c01568814306677f85cdedf0363c0a56675 100644 (file)
--- a/tests/test_psycopg_dbapi20.py
+++ b/tests/test_psycopg_dbapi20.py
@@ -33,7 +33,7 @@ class PsycopgTests(dbapi20.DatabaseAPI20Test):
 @pytest.mark.usefixtures("with_dsn")
 class PsycopgTPCTests(dbapi20_tpc.TwoPhaseCommitTests):
     driver = psycopg
-    # connect_args = () # set by the fixture
+    connect_args = ()  # set by the fixture
 
     def connect(self):
         return psycopg.connect(*self.connect_args)
@@ -41,7 +41,9 @@ class PsycopgTPCTests(dbapi20_tpc.TwoPhaseCommitTests):
 
 # Shut up warnings
 PsycopgTests.failUnless = PsycopgTests.assertTrue  # type: ignore[assignment]
-PsycopgTPCTests.assertEquals = PsycopgTPCTests.assertEqual
+PsycopgTPCTests.assertEquals = (  # type: ignore[assignment]
+    PsycopgTPCTests.assertEqual
+)
 
 
 @pytest.mark.parametrize(