diff options
Diffstat (limited to 'Lib/test/test_asyncio/test_threads.py')
-rw-r--r-- | Lib/test/test_asyncio/test_threads.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/Lib/test/test_asyncio/test_threads.py b/Lib/test/test_asyncio/test_threads.py index 1138a93e0f7..774380270a7 100644 --- a/Lib/test/test_asyncio/test_threads.py +++ b/Lib/test/test_asyncio/test_threads.py @@ -30,7 +30,9 @@ class ToThreadTests(unittest.IsolatedAsyncioTestCase): func.assert_called_once() async def test_to_thread_concurrent(self): - func = mock.Mock() + calls = [] + def func(): + calls.append(1) futs = [] for _ in range(10): @@ -38,7 +40,7 @@ class ToThreadTests(unittest.IsolatedAsyncioTestCase): futs.append(fut) await asyncio.gather(*futs) - self.assertEqual(func.call_count, 10) + self.assertEqual(sum(calls), 10) async def test_to_thread_args_kwargs(self): # Unlike run_in_executor(), to_thread() should directly accept kwargs. |