Declare type variable for Counter in pool tests

We use the comment-style for type annotation because collections.Counter
is generic only since Python 3.9.

diff --git a/tests/pool/test_pool.py b/tests/pool/test_pool.py
index 47a17a252a852bdb9e3c9e524ef7d7a3df4503de..fc30c973042036bf32606b956313739ed737918a 100644 (file)
--- a/tests/pool/test_pool.py
+++ b/tests/pool/test_pool.py
@@ -818,7 +818,7 @@ def test_uniform_use(dsn, retries):
     for retry in retries:
         with retry:
             with pool.ConnectionPool(dsn, min_size=4) as p:
-                counts = Counter()
+                counts = Counter()  # type: Counter[int]
                 for i in range(8):
                     with p.connection() as conn:
                         sleep(0.1)

diff --git a/tests/pool/test_pool_async.py b/tests/pool/test_pool_async.py
index d2542d3137f307d588398c617f408c1c61303c01..9ce51512ea7be8bc559fabd2878643b6b3f1d259 100644 (file)
--- a/tests/pool/test_pool_async.py
+++ b/tests/pool/test_pool_async.py
@@ -812,7 +812,7 @@ async def test_uniform_use(dsn, retries):
     async for retry in retries:
         with retry:
             async with pool.AsyncConnectionPool(dsn, min_size=4) as p:
-                counts = Counter()
+                counts = Counter()  # type: Counter[int]
                 for i in range(8):
                     async with p.connection() as conn:
                         await asyncio.sleep(0.1)