Skip to content

Commit 87e0b83

Browse files
committed
update scratch file
1 parent 8f87977 commit 87e0b83

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

scratch.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,14 @@
55
import pytest
66

77
import zarr
8-
from zarr.core.group import members_v3
8+
from zarr.core.group import iter_members, members_recursive, iter_members_deep
99
from zarr.storage import MemoryStore
1010
from zarr.testing.store import LatencyStore
1111

1212

13-
@pytest.mark.parametrize("num_members", [4, 8, 16])
13+
@pytest.mark.parametrize("num_members", [10, 100, 1000])
1414
@pytest.mark.parametrize("method", ["default", "fast_members"])
15-
def test_collect_members(num_members: int, method: Literal["fast_members", "default"]) -> None:
15+
def test_collect_members(num_members: int, method: Literal["fast_members", "default", "fast_members_2"]) -> None:
1616
local_store = MemoryStore(mode="a")
1717
local_latency_store = LatencyStore(local_store, get_latency=0.1, set_latency=0.0)
1818

@@ -21,14 +21,10 @@ def test_collect_members(num_members: int, method: Literal["fast_members", "defa
2121

2222
for i in range(num_members):
2323
subgroup = root_group_raw.create_group(f"group_outer_{i}")
24-
for j in range(num_members):
25-
subgroup.create_group(f"group_inner_{j}")
2624

2725
if method == "fast_members":
28-
2926
async def amain() -> None:
30-
res = await members_v3(local_latency_store, path="")
31-
print(res)
27+
res = [x async for x in iter_members(root_group_latency._async_group)]
3228

3329
start = time()
3430
asyncio.run(amain())

0 commit comments

Comments
 (0)