Skip to content

Commit aecbd0f

Browse files
committed
Update benchmark
1 parent 5f99eea commit aecbd0f

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

python/tests/test_benchmark.py

+10-8
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from concurrent.futures import ThreadPoolExecutor
1+
from concurrent.futures import ThreadPoolExecutor, as_completed
22

33
import pytest
44
from pytest_benchmark.fixture import BenchmarkFixture
@@ -7,17 +7,19 @@
77

88

99
def do_work(client: Client):
10-
def delete(path: str):
11-
client.delete(path)
10+
def func(path: str):
11+
client.create(path).close()
12+
return client.delete(path)
1213

1314
with ThreadPoolExecutor(100) as executor:
14-
for i in range(100):
15-
executor.submit(delete, f"/bench{i}")
15+
futures = []
16+
for i in range(1000):
17+
futures.append(executor.submit(func, f"/bench{i}"))
18+
19+
for future in as_completed(futures):
20+
assert future.result()
1621

1722

1823
@pytest.mark.benchmark
1924
def test_threading(client: Client, benchmark: BenchmarkFixture):
20-
for i in range(100):
21-
client.create(f"/bench{i}").close()
22-
2325
benchmark(do_work, client)

0 commit comments

Comments
 (0)