Swift3 颜色转换成图片Image

Swift3下的转换写法：

/// 将颜色转换为图片
    ///
    /// - Parameter color: <#color description#>
    /// - Returns: <#return value description#>
    class func getImageWithColor(color:UIColor)->UIImage{
        let rect = CGRect(x: 0, y: 0,  1, height: 1)
        UIGraphicsBeginImageContext(rect.size)
        let context = UIGraphicsGetCurrentContext()
        context!.setFillColor(color.cgColor)
        context!.fill(rect)
        let image = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        return image!
    }

可直接用方法，或加到Image的类别里面，推荐第二种。