Skip to content

Commit 3341173

Browse files
fix(benchmark): adapt callback benchmark to lib conditions (aio-libs#710)
1 parent a363fb8 commit 3341173

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

benchmark.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import asyncio
2+
from functools import partial
23
from typing import Any, Callable
34

45
import pytest
@@ -305,9 +306,10 @@ async def dummy_coro():
305306

306307
iterations = range(1000)
307308
create_future = loop.create_future
308-
callback = func._task_done_callback
309+
callback_fn = func._task_done_callback
309310

310311
@benchmark
311312
def run() -> None:
312313
for i in iterations:
313-
callback(create_future(), i, task)
314+
callback = partial(callback_fn, create_future(), i)
315+
callback(task)

0 commit comments

Comments
 (0)