Skip to content

Commit 8c65365

Browse files
committed
fix typing in test
1 parent 60d6933 commit 8c65365

File tree

1 file changed

+16
-8
lines changed

1 file changed

+16
-8
lines changed

numcodecs/tests/test_zarr3.py

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -317,17 +317,25 @@ def test_cast_numcodecs_to_v3(store: Store, codec_v2, expected_v3_cls) -> None:
317317
assert result_v3.__class__ == expected_v3_cls
318318
assert result_v3.codec_config == codec_v2.get_config()
319319

320-
from zarr.core.array import CompressorsLike, FiltersLike, SerializerLike
321-
322-
codec_args: FiltersLike | SerializerLike | CompressorsLike
323-
320+
filters = "auto"
321+
serializer = "auto"
322+
compressors = "auto"
324323
if issubclass(expected_v3_cls, numcodecs.zarr3._NumcodecsArrayArrayCodec):
325-
codec_args = {"filters": [result_v3]}
324+
filters = [result_v3]
326325
elif issubclass(expected_v3_cls, numcodecs.zarr3._NumcodecsArrayBytesCodec):
327-
codec_args = {"serializer": result_v3}
326+
serializer = result_v3
328327
elif issubclass(expected_v3_cls, numcodecs.zarr3._NumcodecsBytesBytesCodec):
329-
codec_args = {"compressors": [result_v3]}
328+
compressors = [result_v3]
330329
else:
331330
raise TypeError(f"unsupported type: {expected_v3_cls}")
332331

333-
zarr.create_array(store, shape=(64,), chunks=(64,), dtype=np.bool, fill_value=0, **codec_args)
332+
zarr.create_array(
333+
store,
334+
shape=(64,),
335+
chunks=(64,),
336+
dtype=np.bool,
337+
fill_value=0,
338+
filters=filters,
339+
compressors=compressors,
340+
serializer=serializer,
341+
)

0 commit comments

Comments
 (0)