mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-27 18:36:17 +03:00
83 lines
3.1 KiB
Python
83 lines
3.1 KiB
Python
|
from tester import *
|
||
|
|
|
||
|
|
from PIL import Image
|
||
|
|
try:
|
||
|
|
from PIL import ImageCms
|
||
|
|
except ImportError:
|
||
|
|
skip()
|
||
|
|
|
||
|
|
SRGB = "Tests/icc/sRGB.icm"
|
||
|
|
|
||
|
|
def test_sanity():
|
||
|
|
|
||
|
|
# basic smoke test.
|
||
|
|
# this mostly follows the cms_test outline.
|
||
|
|
|
||
|
|
v = ImageCms.versions() # should return four strings
|
||
|
|
assert_equal(v[0], '0.1.0 pil')
|
||
|
|
assert_equal(list(map(type, v)), [str, str, str, str])
|
||
|
|
|
||
|
|
# internal version number
|
||
|
|
assert_match(ImageCms.core.littlecms_version, "\d+\.\d+$")
|
||
|
|
|
||
|
|
i = ImageCms.profileToProfile(lena(), SRGB, SRGB)
|
||
|
|
assert_image(i, "RGB", (128, 128))
|
||
|
|
|
||
|
|
t = ImageCms.buildTransform(SRGB, SRGB, "RGB", "RGB")
|
||
|
|
i = ImageCms.applyTransform(lena(), t)
|
||
|
|
assert_image(i, "RGB", (128, 128))
|
||
|
|
|
||
|
|
p = ImageCms.createProfile("sRGB")
|
||
|
|
o = ImageCms.getOpenProfile(SRGB)
|
||
|
|
t = ImageCms.buildTransformFromOpenProfiles(p, o, "RGB", "RGB")
|
||
|
|
i = ImageCms.applyTransform(lena(), t)
|
||
|
|
assert_image(i, "RGB", (128, 128))
|
||
|
|
|
||
|
|
t = ImageCms.buildProofTransform(SRGB, SRGB, SRGB, "RGB", "RGB")
|
||
|
|
assert_equal(t.inputMode, "RGB")
|
||
|
|
assert_equal(t.outputMode, "RGB")
|
||
|
|
i = ImageCms.applyTransform(lena(), t)
|
||
|
|
assert_image(i, "RGB", (128, 128))
|
||
|
|
|
||
|
|
# get profile information for file
|
||
|
|
assert_equal(ImageCms.getProfileName(SRGB).strip(),
|
||
|
|
'IEC 61966-2.1 Default RGB colour space - sRGB')
|
||
|
|
assert_equal(ImageCms.getProfileInfo(SRGB).splitlines(),
|
||
|
|
['sRGB IEC61966-2.1', '',
|
||
|
|
'Copyright (c) 1998 Hewlett-Packard Company', '',
|
||
|
|
'WhitePoint : D65 (daylight)', '',
|
||
|
|
'Tests/icc/sRGB.icm'])
|
||
|
|
assert_equal(ImageCms.getDefaultIntent(SRGB), 0)
|
||
|
|
assert_equal(ImageCms.isIntentSupported(
|
||
|
|
SRGB, ImageCms.INTENT_ABSOLUTE_COLORIMETRIC,
|
||
|
|
ImageCms.DIRECTION_INPUT), 1)
|
||
|
|
|
||
|
|
# same, using profile object
|
||
|
|
p = ImageCms.createProfile("sRGB")
|
||
|
|
assert_equal(ImageCms.getProfileName(p).strip(),
|
||
|
|
'sRGB built-in - (lcms internal)')
|
||
|
|
assert_equal(ImageCms.getProfileInfo(p).splitlines(),
|
||
|
|
['sRGB built-in', '', 'WhitePoint : D65 (daylight)', '', ''])
|
||
|
|
assert_equal(ImageCms.getDefaultIntent(p), 0)
|
||
|
|
assert_equal(ImageCms.isIntentSupported(
|
||
|
|
p, ImageCms.INTENT_ABSOLUTE_COLORIMETRIC,
|
||
|
|
ImageCms.DIRECTION_INPUT), 1)
|
||
|
|
|
||
|
|
# extensions
|
||
|
|
i = Image.open("Tests/images/rgb.jpg")
|
||
|
|
p = ImageCms.getOpenProfile(BytesIO(i.info["icc_profile"]))
|
||
|
|
assert_equal(ImageCms.getProfileName(p).strip(),
|
||
|
|
'IEC 61966-2.1 Default RGB colour space - sRGB')
|
||
|
|
|
||
|
|
# the procedural pyCMS API uses PyCMSError for all sorts of errors
|
||
|
|
assert_exception(ImageCms.PyCMSError, lambda: ImageCms.profileToProfile(lena(), "foo", "bar"))
|
||
|
|
assert_exception(ImageCms.PyCMSError, lambda: ImageCms.buildTransform("foo", "bar", "RGB", "RGB"))
|
||
|
|
assert_exception(ImageCms.PyCMSError, lambda: ImageCms.getProfileName(None))
|
||
|
|
assert_exception(ImageCms.PyCMSError, lambda: ImageCms.isIntentSupported(SRGB, None, None))
|
||
|
|
|
||
|
|
# test PointTransform convenience API
|
||
|
|
im = lena().point(t)
|
||
|
|
|
||
|
|
# try fetching the profile for the current display device
|
||
|
|
assert_no_exception(lambda: ImageCms.get_display_profile())
|