Pillow/Tests/test_format_hsv.py

137 lines
3.2 KiB
Python
Raw Normal View History

import colorsys
import itertools
2014-07-22 09:36:40 +04:00
from PIL import Image
2020-02-25 12:57:27 +03:00
from .helper import assert_image_similar, hopper
2014-08-28 15:44:19 +04:00
2014-07-22 09:36:40 +04:00
2020-02-25 12:57:27 +03:00
def int_to_float(i):
return i / 255
2014-08-28 15:44:19 +04:00
2020-02-25 12:57:27 +03:00
def str_to_float(i):
return ord(i) / 255
2014-08-28 15:44:19 +04:00
2014-07-22 09:36:40 +04:00
2020-02-25 12:57:27 +03:00
def tuple_to_ints(tp):
x, y, z = tp
return int(x * 255.0), int(y * 255.0), int(z * 255.0)
2014-07-22 09:36:40 +04:00
2014-08-28 15:44:19 +04:00
2020-02-25 12:57:27 +03:00
def test_sanity():
Image.new("HSV", (100, 100))
def wedge():
w = Image._wedge()
w90 = w.rotate(90)
(px, h) = w.size
r = Image.new("L", (px * 3, h))
g = r.copy()
b = r.copy()
r.paste(w, (0, 0))
r.paste(w90, (px, 0))
g.paste(w90, (0, 0))
g.paste(w, (2 * px, 0))
b.paste(w, (px, 0))
b.paste(w90, (2 * px, 0))
img = Image.merge("RGB", (r, g, b))
return img
def to_xxx_colorsys(im, func, mode):
# convert the hard way using the library colorsys routines.
(r, g, b) = im.split()
conv_func = int_to_float
converted = [
tuple_to_ints(func(conv_func(_r), conv_func(_g), conv_func(_b)))
for (_r, _g, _b) in itertools.zip_longest(r.tobytes(), g.tobytes(), b.tobytes())
]
new_bytes = b"".join(
bytes(chr(h) + chr(s) + chr(v), "latin-1") for (h, s, v) in converted
)
hsv = Image.frombytes(mode, r.size, new_bytes)
return hsv
def to_hsv_colorsys(im):
return to_xxx_colorsys(im, colorsys.rgb_to_hsv, "HSV")
def to_rgb_colorsys(im):
return to_xxx_colorsys(im, colorsys.hsv_to_rgb, "RGB")
def test_wedge():
src = wedge().resize((3 * 32, 32), Image.BILINEAR)
im = src.convert("HSV")
comparable = to_hsv_colorsys(src)
assert_image_similar(
im.getchannel(0), comparable.getchannel(0), 1, "Hue conversion is wrong"
)
assert_image_similar(
im.getchannel(1), comparable.getchannel(1), 1, "Saturation conversion is wrong",
)
assert_image_similar(
im.getchannel(2), comparable.getchannel(2), 1, "Value conversion is wrong"
)
comparable = src
im = im.convert("RGB")
assert_image_similar(
im.getchannel(0), comparable.getchannel(0), 3, "R conversion is wrong"
)
assert_image_similar(
im.getchannel(1), comparable.getchannel(1), 3, "G conversion is wrong"
)
assert_image_similar(
im.getchannel(2), comparable.getchannel(2), 3, "B conversion is wrong"
)
def test_convert():
im = hopper("RGB").convert("HSV")
comparable = to_hsv_colorsys(hopper("RGB"))
assert_image_similar(
im.getchannel(0), comparable.getchannel(0), 1, "Hue conversion is wrong"
)
assert_image_similar(
im.getchannel(1), comparable.getchannel(1), 1, "Saturation conversion is wrong",
)
assert_image_similar(
im.getchannel(2), comparable.getchannel(2), 1, "Value conversion is wrong"
)
def test_hsv_to_rgb():
comparable = to_hsv_colorsys(hopper("RGB"))
converted = comparable.convert("RGB")
comparable = to_rgb_colorsys(comparable)
assert_image_similar(
converted.getchannel(0), comparable.getchannel(0), 3, "R conversion is wrong",
)
assert_image_similar(
converted.getchannel(1), comparable.getchannel(1), 3, "G conversion is wrong",
)
assert_image_similar(
converted.getchannel(2), comparable.getchannel(2), 3, "B conversion is wrong",
)