Pillow/Tests/test_file_dds.py

98 lines
2.9 KiB
Python
Raw Normal View History

2016-01-16 00:34:36 +03:00
from io import BytesIO
2016-01-06 11:59:37 +03:00
2016-01-16 00:34:36 +03:00
from helper import unittest, PillowTestCase
2016-01-06 11:59:37 +03:00
from PIL import Image, DdsImagePlugin
TEST_FILE_DXT1 = "Tests/images/dxt1-rgb-4bbp-noalpha_MipMaps-1.dds"
2016-01-06 12:59:47 +03:00
TEST_FILE_DXT3 = "Tests/images/dxt3-argb-8bbp-explicitalpha_MipMaps-1.dds"
2016-01-06 11:59:37 +03:00
TEST_FILE_DXT5 = "Tests/images/dxt5-argb-8bbp-interpolatedalpha_MipMaps-1.dds"
2016-01-06 13:26:58 +03:00
class TestFileDds(PillowTestCase):
"""Test DdsImagePlugin"""
2016-01-06 11:59:37 +03:00
def test_sanity_dxt1(self):
2016-01-06 13:26:58 +03:00
"""Check DXT1 images can be opened"""
target = Image.open(TEST_FILE_DXT1.replace('.dds', '.png'))
2016-01-06 11:59:37 +03:00
im = Image.open(TEST_FILE_DXT1)
im.load()
2016-02-05 01:57:13 +03:00
2016-01-06 12:24:09 +03:00
self.assertEqual(im.format, "DDS")
self.assertEqual(im.mode, "RGBA")
2016-01-06 11:59:37 +03:00
self.assertEqual(im.size, (256, 256))
2016-02-05 01:57:13 +03:00
# This target image is from the test set of images, and is exact.
2016-01-08 18:59:18 +03:00
self.assert_image_equal(target.convert('RGBA'), im)
2016-01-06 11:59:37 +03:00
def test_sanity_dxt5(self):
2016-01-06 13:26:58 +03:00
"""Check DXT5 images can be opened"""
target = Image.open(TEST_FILE_DXT5.replace('.dds', '.png'))
2016-01-06 11:59:37 +03:00
im = Image.open(TEST_FILE_DXT5)
im.load()
2016-01-06 11:59:37 +03:00
2016-01-06 12:24:09 +03:00
self.assertEqual(im.format, "DDS")
self.assertEqual(im.mode, "RGBA")
2016-01-06 11:59:37 +03:00
self.assertEqual(im.size, (256, 256))
# Imagemagick, which generated this target image from the .dds
# has a slightly different decoder than is standard. It looks
# a little brighter. The 0,0 pixel is (00,6c,f8,ff) by our code,
# and by the target image for the DXT1, and the imagemagick .png
# is giving (00, 6d, ff, ff). So, assert similar, pretty tight
2016-02-05 01:57:13 +03:00
# I'm currently seeing about a 3 for the epsilon.
self.assert_image_similar(target, im, 5)
2016-01-08 18:59:18 +03:00
2016-01-06 12:59:47 +03:00
def test_sanity_dxt3(self):
2016-01-06 13:26:58 +03:00
"""Check DXT3 images are not supported"""
2016-01-06 12:59:47 +03:00
self.assertRaises(NotImplementedError,
lambda: Image.open(TEST_FILE_DXT3))
2016-01-06 11:59:37 +03:00
def test__validate_true(self):
2016-01-06 13:26:58 +03:00
"""Check valid prefix"""
2016-01-06 11:59:37 +03:00
# Arrange
prefix = b"DDS etc"
# Act
output = DdsImagePlugin._validate(prefix)
# Assert
self.assertTrue(output)
def test__validate_false(self):
2016-01-06 13:26:58 +03:00
"""Check invalid prefix"""
2016-01-06 11:59:37 +03:00
# Arrange
prefix = b"something invalid"
# Act
output = DdsImagePlugin._validate(prefix)
# Assert
self.assertFalse(output)
2016-01-16 00:34:36 +03:00
def test_short_header(self):
""" Check a short header"""
with open(TEST_FILE_DXT5, 'rb') as f:
img_file = f.read()
def short_header():
2016-02-05 01:57:13 +03:00
Image.open(BytesIO(img_file[:119]))
2016-01-16 00:34:36 +03:00
self.assertRaises(IOError, short_header)
def test_short_file(self):
""" Check that the appropriate error is thrown for a short file"""
2016-02-05 01:57:13 +03:00
2016-01-16 00:34:36 +03:00
with open(TEST_FILE_DXT5, 'rb') as f:
img_file = f.read()
def short_file():
2016-02-05 01:57:13 +03:00
Image.open(BytesIO(img_file[:-100]))
2016-01-16 00:34:36 +03:00
self.assertRaises(IOError, short_file)
2016-01-06 11:59:37 +03:00
if __name__ == '__main__':
unittest.main()