def test_dask_2D():
"""Test adding 2D dask image."""
viewer = ViewerModel()
da.random.seed(0)
data = da.random.random((10, 15))
viewer.add_image(data)
assert np.all(viewer.layers[0].data == data)
def test_xarray_nD():
"""Test adding nD xarray image."""
viewer = ViewerModel()
np.random.seed(0)
data = np.random.random((10, 15, 6, 16))
xdata = xr.DataArray(data, dims=['t', 'z', 'y', 'x'])
viewer.add_image(xdata)
assert np.all(viewer.layers[0].data == xdata)
def test_zarr_dask_nD():
"""Test adding nD zarr image."""
viewer = ViewerModel()
data = zarr.zeros((200, 100, 50), chunks=(40, 20, 10))
data[53:63, 10:20, :] = 1
zdata = da.from_zarr(data)
viewer.add_image(zdata)
assert np.all(viewer.layers[0].data == zdata)
def test_zarr_nD():
"""Test adding nD zarr image."""
viewer = ViewerModel()
data = zarr.zeros((200, 100, 50), chunks=(40, 20, 10))
data[53:63, 10:20, :] = 1
# If passing a zarr file directly, must pass contrast_limits
viewer.add_image(data, contrast_limits=[0, 1])
assert np.all(viewer.layers[0].data == data)
def test_receive_return_proxy_object():
"""Test that an"""
viewer = ViewerModel()
viewer.add_image(np.random.random((20, 20)))
pv = PublicOnlyProxy(viewer)
# simulates behavior from outside of napari
with patch('napari.utils.misc.ROOT_DIR', new='/some/other/package'):
# the recursion means this layer will be a Proxy Object on __getitem__
layer = pv.layers[-1]
assert isinstance(layer, PublicOnlyProxy)
# remove and add it back, should be fine
add_layer = getattr(pv, 'add_layer')
add_layer(layer)
assert len(viewer.layers) == 2
def test_image_dytpes(dtype):
"""Test different dtype images."""
np.random.seed(0)
viewer = ViewerModel()
# add dtype image data
data = np.random.randint(20, size=(30, 40)).astype(dtype)
viewer.add_image(data)
assert np.all(viewer.layers[0].data == data)
# add dtype multiscale data
data = [
np.random.randint(20, size=(30, 40)).astype(dtype),
np.random.randint(20, size=(15, 20)).astype(dtype),
]
viewer.add_image(data, multiscale=True)
assert np.all(viewer.layers[1].data == data)