Ejemplo n.º 1
0
def test_colormap():
    """Test colormap."""
    np.random.seed(0)
    data = np.random.randint(20, size=(10, 15))
    layer = Labels(data)
    assert isinstance(layer.colormap, Colormap)
    assert layer.colormap.name == 'label_colormap'

    layer.new_colormap()
    assert isinstance(layer.colormap, Colormap)
    assert layer.colormap.name == 'label_colormap'
Ejemplo n.º 2
0
def test_colormap():
    """Test colormap."""
    np.random.seed(0)
    data = np.random.randint(20, size=(10, 15))
    layer = Labels(data)
    assert type(layer.colormap) == tuple
    assert layer.colormap[0] == 'random'
    assert type(layer.colormap[1]) == Colormap

    layer.new_colormap()
    assert type(layer.colormap) == tuple
    assert layer.colormap[0] == 'random'
    assert type(layer.colormap[1]) == Colormap