55import pytest
66
77import zarr
8- from zarr .core .group import members_v3
8+ from zarr .core .group import iter_members , members_recursive , iter_members_deep
99from zarr .storage import MemoryStore
1010from 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