def test_to_labels():
"""Test the labels generation."""
shape = (10, 4, 2)
np.random.seed(0)
data = 20 * np.random.random(shape)
layer = Shapes(data)
labels = layer.to_labels()
assert labels.ndim == 2
assert len(np.unique(labels)) <= 11
labels = layer.to_labels(labels_shape=[20, 20])
assert labels.shape == (20, 20)
assert len(np.unique(labels)) <= 11
def test_to_labels_3D():
"""Test label generation for 3D data"""
data = [
[[0, 100, 100], [0, 100, 200], [0, 200, 200], [0, 200, 100]],
[[1, 125, 125], [1, 125, 175], [1, 175, 175], [1, 175, 125]],
[[2, 100, 100], [2, 100, 200], [2, 200, 200], [2, 200, 100]],
]
labels_shape = (3, 300, 300)
layer = Shapes(np.array(data), shape_type='polygon')
labels = layer.to_labels(labels_shape=labels_shape)
assert np.all(labels.shape == labels_shape)
assert np.all(np.unique(labels) == [0, 1, 2, 3])
