Fixed database typo and removed unnecessary class identifier.

venv/Lib/site-packages/skimage/draw/tests/test_random_shapes.py

@@ -0,0 +1,171 @@
+import numpy as np
+
+from skimage.draw import random_shapes
+
+from skimage._shared import testing
+from skimage._shared._warnings import expected_warnings
+
+
+def test_generates_color_images_with_correct_shape():
+    image, _ = random_shapes((128, 128), max_shapes=10)
+    assert image.shape == (128, 128, 3)
+
+
+def test_generates_gray_images_with_correct_shape():
+    image, _ = random_shapes(
+        (4567, 123), min_shapes=3, max_shapes=20, multichannel=False)
+    assert image.shape == (4567, 123)
+
+
+def test_generates_correct_bounding_boxes_for_rectangles():
+    image, labels = random_shapes(
+        (128, 128),
+        max_shapes=1,
+        shape='rectangle',
+        random_seed=42)
+    assert len(labels) == 1
+    label, bbox = labels[0]
+    assert label == 'rectangle', label
+
+    crop = image[bbox[0][0]:bbox[0][1], bbox[1][0]:bbox[1][1]]
+
+    # The crop is filled.
+    assert (crop >= 0).all() and (crop < 255).all()
+
+    # The crop is complete.
+    image[bbox[0][0]:bbox[0][1], bbox[1][0]:bbox[1][1]] = 255
+    assert (image == 255).all()
+
+
+def test_generates_correct_bounding_boxes_for_triangles():
+    image, labels = random_shapes(
+        (128, 128),
+        max_shapes=1,
+        shape='triangle',
+        random_seed=42)
+    assert len(labels) == 1
+    label, bbox = labels[0]
+    assert label == 'triangle', label
+
+    crop = image[bbox[0][0]:bbox[0][1], bbox[1][0]:bbox[1][1]]
+
+    # The crop is filled.
+    assert (crop >= 0).any() and (crop < 255).any()
+
+    # The crop is complete.
+    image[bbox[0][0]:bbox[0][1], bbox[1][0]:bbox[1][1]] = 255
+    assert (image == 255).all()
+
+
+def test_generates_correct_bounding_boxes_for_circles():
+    image, labels = random_shapes(
+        (43, 44),
+        max_shapes=1,
+        min_size=20,
+        max_size=20,
+        shape='circle',
+        random_seed=42)
+    assert len(labels) == 1
+    label, bbox = labels[0]
+    assert label == 'circle', label
+
+    crop = image[bbox[0][0]:bbox[0][1], bbox[1][0]:bbox[1][1]]
+
+    # The crop is filled.
+    assert (crop >= 0).any() and (crop < 255).any()
+
+    # The crop is complete.
+    image[bbox[0][0]:bbox[0][1], bbox[1][0]:bbox[1][1]] = 255
+    assert (image == 255).all()
+
+
+def test_generates_correct_bounding_boxes_for_ellipses():
+    image, labels = random_shapes(
+        (43, 44),
+        max_shapes=1,
+        min_size=20,
+        max_size=20,
+        shape='ellipse',
+        random_seed=42)
+    assert len(labels) == 1
+    label, bbox = labels[0]
+    assert label == 'ellipse', label
+
+    crop = image[bbox[0][0]:bbox[0][1], bbox[1][0]:bbox[1][1]]
+
+    # The crop is filled.
+    assert (crop >= 0).any() and (crop < 255).any()
+
+    # The crop is complete.
+    image[bbox[0][0]:bbox[0][1], bbox[1][0]:bbox[1][1]] = 255
+    assert (image == 255).all()
+
+
+def test_generate_circle_throws_when_size_too_small():
+    with testing.raises(ValueError):
+        random_shapes(
+            (64, 128), max_shapes=1, min_size=1, max_size=1, shape='circle')
+
+
+def test_generate_ellipse_throws_when_size_too_small():
+    with testing.raises(ValueError):
+        random_shapes(
+            (64, 128), max_shapes=1, min_size=1, max_size=1, shape='ellipse')
+
+
+def test_generate_triangle_throws_when_size_too_small():
+    with testing.raises(ValueError):
+        random_shapes(
+            (128, 64), max_shapes=1, min_size=1, max_size=1, shape='triangle')
+
+
+def test_can_generate_one_by_one_rectangle():
+    image, labels = random_shapes(
+        (50, 128),
+        max_shapes=1,
+        min_size=1,
+        max_size=1,
+        shape='rectangle')
+    assert len(labels) == 1
+    _, bbox = labels[0]
+    crop = image[bbox[0][0]:bbox[0][1], bbox[1][0]:bbox[1][1]]
+
+    # rgb
+    assert (np.shape(crop) == (1, 1, 3) and np.any(crop >= 1)
+            and np.any(crop < 255))
+
+
+def test_throws_when_intensity_range_out_of_range():
+    with testing.raises(ValueError):
+        random_shapes((1000, 1234), max_shapes=1, multichannel=False,
+                      intensity_range=(0, 256))
+    with testing.raises(ValueError):
+        random_shapes((2, 2), max_shapes=1,
+                      intensity_range=((-1, 255),))
+
+
+def test_returns_empty_labels_and_white_image_when_cannot_fit_shape():
+    # The circle will never fit this.
+    with expected_warnings(['Could not fit']):
+        image, labels = random_shapes(
+            (10000, 10000), max_shapes=1, min_size=10000, shape='circle')
+    assert len(labels) == 0
+    assert (image == 255).all()
+
+
+def test_random_shapes_is_reproducible_with_seed():
+    random_seed = 42
+    labels = []
+    for _ in range(5):
+        _, label = random_shapes((128, 128), max_shapes=5,
+                                 random_seed=random_seed)
+        labels.append(label)
+    assert all(other == labels[0] for other in labels[1:])
+
+
+def test_generates_white_image_when_intensity_range_255():
+    image, labels = random_shapes((128, 128), max_shapes=3,
+                                  intensity_range=((255, 255),),
+                                  random_seed=42)
+    assert len(labels) > 0
+    assert (image == 255).all()