def test_add_edge_color_cycle_to_empty_layer():
""" Test adding a point to an empty layer when edge color is a color cycle
See: https://github.com/napari/napari/pull/1069
"""
default_properties = {'point_type': np.array(['A'])}
color_cycle = ['red', 'blue']
layer = Points(
properties=default_properties,
edge_color='point_type',
edge_color_cycle=color_cycle,
)
# verify the current_edge_color is correct
edge_color = transform_color(color_cycle[0])
assert np.all(layer._current_edge_color == edge_color)
# add a point
layer.add([10, 10])
props = {'point_type': np.array(['A'])}
edge_color = np.array([[1, 0, 0, 1]])
assert layer.properties == props
np.testing.assert_allclose(layer.edge_color, edge_color)
# add a point with a new property
layer.selected_data = []
layer.current_properties = {'point_type': np.array(['B'])}
layer.add([12, 12])
new_color = np.array([0, 0, 1, 1])
edge_color = np.vstack((edge_color, new_color))
new_properties = {'point_type': np.array(['A', 'B'])}
np.testing.assert_allclose(layer.edge_color, edge_color)
np.testing.assert_equal(layer.properties, new_properties)
def test_color_cycle(attribute, color_cycle):
"""Test setting edge/face color with a color cycle list"""
# create Points using list color cycle
shape = (10, 2)
np.random.seed(0)
data = 20 * np.random.random(shape)
properties = {'point_type': _make_cycled_properties(['A', 'B'], shape[0])}
points_kwargs = {
'properties': properties,
f'{attribute}_color': 'point_type',
f'{attribute}_color_cycle': color_cycle,
}
layer = Points(data, **points_kwargs)
assert layer.properties == properties
color_array = transform_color(
list(islice(cycle(color_cycle), 0, shape[0]))
)
layer_color = getattr(layer, f'{attribute}_color')
np.testing.assert_allclose(layer_color, color_array)
# Add new point and test its color
coord = [18, 18]
layer.selected_data = {0}
layer.add(coord)
layer_color = getattr(layer, f'{attribute}_color')
assert len(layer_color) == shape[0] + 1
np.testing.assert_allclose(
layer_color,
np.vstack((color_array, transform_color('red'))),
)
# Check removing data adjusts colors correctly
layer.selected_data = {0, 2}
layer.remove_selected()
assert len(layer.data) == shape[0] - 1
layer_color = getattr(layer, f'{attribute}_color')
assert len(layer_color) == shape[0] - 1
np.testing.assert_allclose(
layer_color,
np.vstack((color_array[1], color_array[3:], transform_color('red'))),
)
# refresh colors
layer.refresh_colors(update_color_mapping=True)
# test adding a point with a new property value
layer.selected_data = {}
current_properties = layer.current_properties
current_properties['point_type'] = np.array(['new'])
layer.current_properties = current_properties
layer.add([10, 10])
color_cycle_map = getattr(layer, f'{attribute}_color_cycle_map')
assert 'new' in color_cycle_map
np.testing.assert_allclose(
color_cycle_map['new'], np.squeeze(transform_color(color_cycle[0]))
)
def test_updating_points_properties():
# test adding points initialized with properties
shape = (10, 2)
np.random.seed(0)
data = 20 * np.random.random(shape)
properties = {'point_type': _make_cycled_properties(['A', 'B'], shape[0])}
layer = Points(data, properties=copy(properties))
layer.mode = 'select'
layer.selected_data = [len(data) - 1]
layer.current_properties = {'point_type': np.array(['A'])}
updated_properties = properties
updated_properties['point_type'][-1] = 'A'
np.testing.assert_equal(layer.properties, updated_properties)
def test_add_color_cycle_to_empty_layer(attribute):
"""Test adding a point to an empty layer when edge/face color is a color cycle
See: https://github.com/napari/napari/pull/1069
"""
default_properties = {'point_type': np.array(['A'])}
color_cycle = ['red', 'blue']
points_kwargs = {
'properties': default_properties,
f'{attribute}_color': 'point_type',
f'{attribute}_color_cycle': color_cycle,
}
layer = Points(**points_kwargs)
# verify the current_edge_color is correct
expected_color = transform_color(color_cycle[0])[0]
color_manager = getattr(layer, f'_{attribute}')
current_color = color_manager.current_color
np.testing.assert_allclose(current_color, expected_color)
# add a point
layer.add([10, 10])
props = {'point_type': np.array(['A'])}
expected_color = np.array([[1, 0, 0, 1]])
np.testing.assert_equal(layer.properties, props)
attribute_color = getattr(layer, f'{attribute}_color')
np.testing.assert_allclose(attribute_color, expected_color)
# add a point with a new property
layer.selected_data = []
layer.current_properties = {'point_type': np.array(['B'])}
layer.add([12, 12])
new_color = np.array([0, 0, 1, 1])
expected_color = np.vstack((expected_color, new_color))
new_properties = {'point_type': np.array(['A', 'B'])}
attribute_color = getattr(layer, f'{attribute}_color')
np.testing.assert_allclose(attribute_color, expected_color)
np.testing.assert_equal(layer.properties, new_properties)
