@@ -331,9 +331,12 @@ def test_pickle() -> None:
331331 codec = ShardingCodec (chunk_shape = (8 , 8 ))
332332 assert pickle .loads (pickle .dumps (codec )) == codec
333333
334+
334335@pytest .mark .parametrize ("store" , ("local" , "memory" ), indirect = ["store" ])
335- @pytest .mark .parametrize ("index_location" , [ShardingCodecIndexLocation .start , ShardingCodecIndexLocation .end ])
336- async def test_sharding_with_empty_inner_chunk (store : Store , index_location ):
336+ @pytest .mark .parametrize (
337+ "index_location" , [ShardingCodecIndexLocation .start , ShardingCodecIndexLocation .end ]
338+ )
339+ async def test_sharding_with_empty_inner_chunk (store : Store , index_location : ShardingCodecIndexLocation ) -> None :
337340 data = np .arange (0 , 16 * 16 , dtype = "uint32" ).reshape ((16 , 16 ))
338341 fill_value = 1
339342
@@ -345,7 +348,7 @@ async def test_sharding_with_empty_inner_chunk(store: Store, index_location):
345348 chunk_shape = (8 , 8 ),
346349 dtype = "uint32" ,
347350 fill_value = fill_value ,
348- codecs = [ShardingCodec (chunk_shape = (4 , 4 ), index_location = index_location , index_codecs = [ BytesCodec ()] )],
351+ codecs = [ShardingCodec (chunk_shape = (4 , 4 ), index_location = index_location )],
349352 )
350353 data [:4 , :4 ] = fill_value
351354 await a .setitem (..., data )
0 commit comments