从函数中的数组返回项

Question

我有一个函数，我想返回一个UIImage，但是这个UIImage必须在我创建的UIImages数组中。

这是可行的，但如果项目不是数组的一部分，我希望它失败。

private let screenImages: [UIImage] = [#imageLiteral(resourceName: "screen-1"), #imageLiteral(resourceName: "screen-2"), #imageLiteral(resourceName: "screen-3")]

private let getImage() -> (UIImage) {

    let random = Int(arc4random_uniform(UInt32(screenImages.count)))
    let image = screenImages[random]
    return image

}

所以在函数的转义过程中，我能够告诉函数应该返回什么样的图像，而不是UIImage。像这样的东西

private let getImage() -> (UIImage in screenImages)