def test_adding_properties(attribute):
"""Test adding properties to an existing layer"""
shape = (10, 4, 2)
np.random.seed(0)
data = 20 * np.random.random(shape)
layer = Shapes(data)
# add properties
properties = {'shape_type': _make_cycled_properties(['A', 'B'], shape[0])}
layer.properties = properties
np.testing.assert_equal(layer.properties, properties)
# add properties as a dataframe
properties_df = pd.DataFrame(properties)
layer.properties = properties_df
np.testing.assert_equal(layer.properties, properties)
# add properties as a dictionary with list values
properties_list = {
'shape_type': list(_make_cycled_properties(['A', 'B'], shape[0]))
}
layer.properties = properties_list
assert isinstance(layer.properties['shape_type'], np.ndarray)
# removing a property that was the _*_color_property should give a warning
setattr(layer, f'_{attribute}_color_property', 'shape_type')
properties_2 = {
'not_shape_type': _make_cycled_properties(['A', 'B'], shape[0])
}
with pytest.warns(RuntimeWarning):
layer.properties = properties_2
def test_change_properties_updates_node_strings():
shapes = np.random.rand(3, 4, 2)
properties = {'class': np.array(['A', 'B', 'C'])}
layer = Shapes(shapes, properties=properties, text='class')
vispy_layer = VispyShapesLayer(layer)
text_node = vispy_layer._get_text_node()
np.testing.assert_array_equal(text_node.text, ['A', 'B', 'C'])
layer.properties = {'class': np.array(['D', 'E', 'F'])}
np.testing.assert_array_equal(text_node.text, ['D', 'E', 'F'])
def test_refresh_text():
"""Test refreshing the text after setting new properties"""
shape = (10, 4, 2)
np.random.seed(0)
data = 20 * np.random.random(shape)
properties = {'shape_type': ['A'] * shape[0]}
layer = Shapes(data, properties=copy(properties), text='shape_type')
new_properties = {'shape_type': ['B'] * shape[0]}
layer.properties = new_properties
np.testing.assert_equal(layer.text.values, new_properties['shape_type'])
