chore: fix mypy >= 1.7 warning in tests

diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py
index c88475a2dd88bf78aca7603a2890fded040cefd1..8d1fb02797ecf5a26487aaeb76bfacb373c53a70 100644 (file)
--- a/tests/test_pipeline.py
+++ b/tests/test_pipeline.py
@@ -589,7 +589,7 @@ def test_concurrency(conn):
         from concurrent.futures import ThreadPoolExecutor
 
         with ThreadPoolExecutor() as e:
-            cursors = e.map(update, values, timeout=len(values))
+            cursors = list(e.map(update, values, timeout=len(values)))
 
         assert sum([cur.fetchone()[0] for cur in cursors]) == sum(values)
 

diff --git a/tests/test_pipeline_async.py b/tests/test_pipeline_async.py
index ee0efccd93987b3e2b672239aea9da817c57072d..b9baa589f4ae92388f0d9cc72547b967fad49585 100644 (file)
--- a/tests/test_pipeline_async.py
+++ b/tests/test_pipeline_async.py
@@ -598,7 +598,7 @@ async def test_concurrency(aconn):
             from concurrent.futures import ThreadPoolExecutor
 
             with ThreadPoolExecutor() as e:
-                cursors = e.map(update, values, timeout=len(values))
+                cursors = list(e.map(update, values, timeout=len(values)))
 
         assert sum([(await cur.fetchone())[0] for cur in cursors]) == sum(values)