1212
1313import zarr .api .asynchronous
1414from zarr import Array , AsyncArray , Group
15- from zarr .codecs import BytesCodec , VLenBytesCodec
15+ from zarr .codecs import BytesCodec , VLenBytesCodec , ZstdCodec
1616from zarr .core ._info import ArrayInfo
1717from zarr .core .array import chunks_initialized
1818from zarr .core .buffer import default_buffer_prototype
@@ -376,7 +376,7 @@ async def test_chunks_initialized() -> None:
376376
377377
378378def test_nbytes_stored () -> None :
379- arr = zarr .create (shape = (100 ,), chunks = (10 ,), dtype = "i4" )
379+ arr = zarr .create (shape = (100 ,), chunks = (10 ,), dtype = "i4" , codecs = [ BytesCodec ()] )
380380 result = arr .nbytes_stored ()
381381 assert result == 366 # the size of the metadata document. This is a fragile test.
382382 arr [:50 ] = 1
@@ -388,7 +388,9 @@ def test_nbytes_stored() -> None:
388388
389389
390390async def test_nbytes_stored_async () -> None :
391- arr = await zarr .api .asynchronous .create (shape = (100 ,), chunks = (10 ,), dtype = "i4" )
391+ arr = await zarr .api .asynchronous .create (
392+ shape = (100 ,), chunks = (10 ,), dtype = "i4" , codecs = [BytesCodec ()]
393+ )
392394 result = await arr .nbytes_stored ()
393395 assert result == 366 # the size of the metadata document. This is a fragile test.
394396 await arr .setitem (slice (50 ), 1 )
@@ -473,13 +475,13 @@ def test_info_v3(self) -> None:
473475 _order = "C" ,
474476 _read_only = False ,
475477 _store_type = "MemoryStore" ,
476- _codecs = [BytesCodec ()],
478+ _codecs = [BytesCodec (), ZstdCodec () ],
477479 _count_bytes = 128 ,
478480 )
479481 assert result == expected
480482
481483 def test_info_complete (self ) -> None :
482- arr = zarr .create (shape = (4 , 4 ), chunks = (2 , 2 ), zarr_format = 3 )
484+ arr = zarr .create (shape = (4 , 4 ), chunks = (2 , 2 ), zarr_format = 3 , codecs = [ BytesCodec ()] )
483485 result = arr .info_complete ()
484486 expected = ArrayInfo (
485487 _zarr_format = 3 ,
@@ -530,13 +532,15 @@ async def test_info_v3_async(self) -> None:
530532 _order = "C" ,
531533 _read_only = False ,
532534 _store_type = "MemoryStore" ,
533- _codecs = [BytesCodec ()],
535+ _codecs = [BytesCodec (), ZstdCodec () ],
534536 _count_bytes = 128 ,
535537 )
536538 assert result == expected
537539
538540 async def test_info_complete_async (self ) -> None :
539- arr = await zarr .api .asynchronous .create (shape = (4 , 4 ), chunks = (2 , 2 ), zarr_format = 3 )
541+ arr = await zarr .api .asynchronous .create (
542+ shape = (4 , 4 ), chunks = (2 , 2 ), zarr_format = 3 , codecs = [BytesCodec ()]
543+ )
540544 result = await arr .info_complete ()
541545 expected = ArrayInfo (
542546 _zarr_format = 3 ,
0 commit comments