Use setattr() when monkeypatching in pool tests

This is to avoid mypy error: Cannot assign to a method.

diff --git a/tests/pool/test_sched.py b/tests/pool/test_sched.py
index d35332fec8812dd2d595bdc9a9751d09c5fec7f5..a3bdd5cfaa074d578589f5ea4e935a25e13921e6 100644 (file)
--- a/tests/pool/test_sched.py
+++ b/tests/pool/test_sched.py
@@ -111,7 +111,7 @@ def test_empty_queue_timeout():
         times.append(time() - t0)
         return rv
 
-    s._event.wait = wait_logging
+    setattr(s._event, "wait", wait_logging)
     s.EMPTY_QUEUE_TIMEOUT = 0.2
 
     t = Thread(target=s.run)
@@ -138,7 +138,7 @@ def test_first_task_rescheduling():
         times.append(time() - t0)
         return rv
 
-    s._event.wait = wait_logging
+    setattr(s._event, "wait", wait_logging)
     s.EMPTY_QUEUE_TIMEOUT = 0.1
 
     s.enter(0.4, lambda: None)

diff --git a/tests/pool/test_sched_async.py b/tests/pool/test_sched_async.py
index 2ea36dd1a310dc7670d5238aac4b1cbd90868ade..651e9038f4c72745b7fb48c77444b8719348c322 100644 (file)
--- a/tests/pool/test_sched_async.py
+++ b/tests/pool/test_sched_async.py
@@ -113,7 +113,7 @@ async def test_empty_queue_timeout():
             times.append(time() - t0)
         return rv
 
-    s._event.wait = wait_logging
+    setattr(s._event, "wait", wait_logging)
     s.EMPTY_QUEUE_TIMEOUT = 0.2
 
     t = create_task(s.run())
@@ -141,7 +141,7 @@ async def test_first_task_rescheduling():
             times.append(time() - t0)
         return rv
 
-    s._event.wait = wait_logging
+    setattr(s._event, "wait", wait_logging)
     s.EMPTY_QUEUE_TIMEOUT = 0.1
 
     async def noop():