mirror of https://github.com/pyodide/pyodide.git
31 lines
743 B
Python
31 lines
743 B
Python
def test_zarr(selenium):
|
|
selenium.load_package(["numpy", "numcodecs", "zarr"])
|
|
selenium.run(
|
|
r"""
|
|
from numcodecs import Blosc
|
|
import numpy as np
|
|
import zarr
|
|
|
|
# basic test
|
|
z = zarr.zeros((1000, 1000), chunks=(100, 100), dtype='i4')
|
|
assert z.shape == (1000, 1000)
|
|
|
|
# test assignment
|
|
z[0, :] = np.arange(1000)
|
|
assert z[0,1] == 1
|
|
|
|
# test saving and loading
|
|
a1 = np.arange(10)
|
|
zarr.save('/tmp/example.zarr', a1)
|
|
a2 = zarr.load('/tmp/example.zarr')
|
|
np.testing.assert_equal(a1, a2)
|
|
|
|
|
|
# test compressor
|
|
compressor = Blosc(cname='zstd', clevel=3, shuffle=Blosc.BITSHUFFLE)
|
|
data = np.arange(10000, dtype='i4').reshape(100, 100)
|
|
z = zarr.array(data, chunks=(10, 10), compressor=compressor)
|
|
assert z.compressor == compressor
|
|
"""
|
|
)
|