lint

Author: TomAugspurger

tests/test_info.py

@@ -1,7 +1,9 @@
 import textwrap
 
+import numpy as np
 import pytest
 
+from zarr.codecs.bytes import BytesCodec
 from zarr.core._info import ArrayInfo, GroupInfo, human_readable_size
 from zarr.core.common import ZarrFormat
 
@@ -51,13 +53,13 @@ def test_group_info_complete(zarr_format: ZarrFormat) -> None:
 def test_array_info(zarr_format: ZarrFormat) -> None:
     info = ArrayInfo(
         _zarr_format=zarr_format,
-        _data_type="int32",
+        _data_type=np.dtype("int32"),
         _shape=(100, 100),
         _chunk_shape=(10, 100),
         _order="C",
         _read_only=True,
         _store_type="MemoryStore",
-        _codecs="[\"BytesCodec(endian=<Endian.little: 'little'>\"]",
+        _codecs=[BytesCodec()],
     )
     result = repr(info)
     assert result == textwrap.dedent(f"""\
@@ -87,13 +89,13 @@ def test_array_info_complete(
     ) = bytes_things
     info = ArrayInfo(
         _zarr_format=zarr_format,
-        _data_type="int32",
+        _data_type=np.dtype("int32"),
         _shape=(100, 100),
         _chunk_shape=(10, 100),
         _order="C",
         _read_only=True,
         _store_type="MemoryStore",
-        _codecs="[\"BytesCodec(endian=<Endian.little: 'little'>\"]",
+        _codecs=[BytesCodec()],
         _count_bytes=count_bytes,
         _count_bytes_stored=count_bytes_stored,
         _count_chunks_initialized=count_chunks_initialized,