Function test_data

dali/test/python/operator_2/test_segmentation_select_masks.py:126–130

()

def test_select_masks_wrong_mask_ids():
    def test_data():
        polygons = [np.array([[0, 0, 2], [1, 3, 5], [2, 6, 8]], dtype=np.int32)]
        vertices = [np.array(np.random.rand(9, 2), dtype=np.float32)]
        mask_ids = [np.array([10, 11], dtype=np.int32)]  # out of bounds ids
        return polygons, vertices, mask_ids

    _test_select_masks_wrong_input(
        lambda: test_data(), err_regex="Selected mask_id .* is not present in the input\\."