def test_LabelledPointUndirectedGraph_add_label():
lgroup = LabelledPointUndirectedGraph(points, adjacency_matrix, mask_dict_2)
new_lgroup = lgroup.add_label('lower2', [0, 1])
assert not is_same_array(new_lgroup.points, lgroup.points)
lower_pcloud = new_lgroup.get_label('lower2')
assert lower_pcloud.n_points == 2
assert_allclose(lower_pcloud.points[0, :], [1., 1., 1.])
assert_allclose(lower_pcloud.points[1, :], [1., 1., 1.])
def test_LabelledPointUndirectedGraph_add_ordered_labels():
lgroup = LabelledPointUndirectedGraph(points, adjacency_matrix, mask_dict_2)
labels = lgroup.labels
assert labels[0] == 'lower'
assert labels[1] == 'upper'
new_lgroup = lgroup.add_label('A', [0, 1])
new_labels = new_lgroup.labels
assert new_labels[2] == 'A'