def test_can_accept_vispy_colormaps():
"""Test that we can accept vispy colormaps."""
colors = np.array([[0, 0, 0, 1], [0, 1, 0, 1], [0, 0, 1, 1]])
vispy_cmap = VispyColormap(colors)
cmap = ensure_colormap(vispy_cmap)
assert isinstance(cmap, Colormap)
np.testing.assert_almost_equal(cmap.colors, colors)
def test_can_accept_napari_colormaps():
"""Test that we can accept napari colormaps."""
colors = np.array([[0, 0, 0, 1], [0, 1, 0, 1], [0, 0, 1, 1]])
napari_cmap = Colormap(colors)
cmap = ensure_colormap(napari_cmap)
assert isinstance(cmap, Colormap)
np.testing.assert_almost_equal(cmap.colors, colors)
def test_can_accept_colormap_dict():
"""Test that we can accept vispy colormaps in a dictionary"""
colors = np.array([[0, 0, 0, 1], [0, 1, 0, 1], [0, 0, 1, 1]])
cmap = ensure_colormap({'colors': colors, 'name': 'special_name'})
assert isinstance(cmap, Colormap)
np.testing.assert_almost_equal(cmap.colors, colors)
assert cmap.name == 'special_name'
def test_can_accept_napari_colormap_name_tuple():
"""Test that we can accept napari colormap named type."""
colors = np.array([[0, 0, 0, 1], [0, 1, 0, 1], [0, 0, 1, 1]])
napari_cmap = Colormap(colors)
cmap = ensure_colormap(('special_name', napari_cmap))
assert isinstance(cmap, Colormap)
np.testing.assert_almost_equal(cmap.colors, colors)
assert cmap.name == 'special_name'
def test_can_accept_napari_colormaps_in_dict():
"""Test that we can accept vispy colormaps in a dictionary"""
colors_a = np.array([[0, 0, 0, 1], [0, 1, 0, 1], [0, 0, 1, 1]])
colors_b = np.array([[0, 0, 0, 1], [1, 0, 0, 1], [0, 0, 1, 1]])
napari_cmap_a = Colormap(colors_a)
napari_cmap_b = Colormap(colors_b)
cmap = ensure_colormap({'a': napari_cmap_a, 'b': napari_cmap_b})
assert isinstance(cmap, Colormap)
np.testing.assert_almost_equal(cmap.colors, colors_a)
assert cmap.name == 'a'
def test_can_accept_named_mpl_colormap():
"""Test we can accept named mpl colormap"""
cmap_name = 'RdYlGn'
cmap = ensure_colormap(cmap_name)
assert isinstance(cmap, Colormap)
assert cmap.name == cmap_name
def test_can_accept_named_vispy_colormaps():
"""Test that we can accept named vispy colormap."""
cmap = ensure_colormap('red')
assert isinstance(cmap, Colormap)
assert cmap.name == 'red'
def test_can_degrade_gracefully():
"""Test that we can degrade gracefully if given something not recognized."""
with pytest.warns(UserWarning):
cmap = ensure_colormap(object)
assert isinstance(cmap, Colormap)
assert cmap.name == 'gray'
def test_ensure_colormap_with_multi_colors(colors):
"""See https://github.com/napari/napari/issues/3141"""
colormap = ensure_colormap(colors)
expected_colors = transform_color(colors)
np.testing.assert_array_equal(colormap.colors, expected_colors)
def test_ensure_colormap_with_single_color(color):
"""See https://github.com/napari/napari/issues/3141"""
colormap = ensure_colormap(color)
np.testing.assert_array_equal(colormap.colors[0], [0, 0, 0, 1])
expected_color = transform_color(color)[0]
np.testing.assert_array_equal(colormap.colors[-1], expected_color)
