import ngff_zarr as nz
import numpy as np

# Generate pixel data (e.g. from a microscope, simulation, etc.)
data = np.random.randint(0, 256, size=(120, 120), dtype=np.uint8)
image = nz.to_ngff_image(data, scale={'y': 1.0, 'x': 1.0})

multiscales = nz.to_multiscales(image, scale_factors=[2, 3, 4])
nz.to_ngff_zarr('example.ome.zarr', multiscales)
image = nz.from_ngff_zarr('example.ome.zarr')
# original ok
assert image.images[0].data.shape == (120, 120)
# factor 2 ok
assert image.images[1].data.shape == (60, 60)
# factor 4 ok 
assert image.images[3].data.shape == (30, 30)
# factor 3 is wrong ==> should be (40, 40) but is (60, 60)
assert image.images[2].data.shape == (40, 40), image.images[2].data.shape