mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-26 18:06:18 +03:00
d50445ff30
Similar to the recent adoption of Black. isort is a Python utility to sort imports alphabetically and automatically separate into sections. By using isort, contributors can quickly and automatically conform to the projects style without thinking. Just let the tool do it. Uses the configuration recommended by the Black to avoid conflicts of style. Rewrite TestImageQt.test_deprecated to no rely on import order.
143 lines
3.9 KiB
Python
143 lines
3.9 KiB
Python
from io import BytesIO
|
|
|
|
from PIL import DdsImagePlugin, Image
|
|
|
|
from .helper import PillowTestCase
|
|
|
|
TEST_FILE_DXT1 = "Tests/images/dxt1-rgb-4bbp-noalpha_MipMaps-1.dds"
|
|
TEST_FILE_DXT3 = "Tests/images/dxt3-argb-8bbp-explicitalpha_MipMaps-1.dds"
|
|
TEST_FILE_DXT5 = "Tests/images/dxt5-argb-8bbp-interpolatedalpha_MipMaps-1.dds"
|
|
TEST_FILE_DX10_BC7 = "Tests/images/bc7-argb-8bpp_MipMaps-1.dds"
|
|
TEST_FILE_UNCOMPRESSED_RGB = "Tests/images/uncompressed_rgb.dds"
|
|
|
|
|
|
class TestFileDds(PillowTestCase):
|
|
"""Test DdsImagePlugin"""
|
|
|
|
def test_sanity_dxt1(self):
|
|
"""Check DXT1 images can be opened"""
|
|
target = Image.open(TEST_FILE_DXT1.replace(".dds", ".png"))
|
|
|
|
im = Image.open(TEST_FILE_DXT1)
|
|
im.load()
|
|
|
|
self.assertEqual(im.format, "DDS")
|
|
self.assertEqual(im.mode, "RGBA")
|
|
self.assertEqual(im.size, (256, 256))
|
|
|
|
self.assert_image_equal(target.convert("RGBA"), im)
|
|
|
|
def test_sanity_dxt5(self):
|
|
"""Check DXT5 images can be opened"""
|
|
|
|
target = Image.open(TEST_FILE_DXT5.replace(".dds", ".png"))
|
|
|
|
im = Image.open(TEST_FILE_DXT5)
|
|
im.load()
|
|
|
|
self.assertEqual(im.format, "DDS")
|
|
self.assertEqual(im.mode, "RGBA")
|
|
self.assertEqual(im.size, (256, 256))
|
|
|
|
self.assert_image_equal(target, im)
|
|
|
|
def test_sanity_dxt3(self):
|
|
"""Check DXT3 images can be opened"""
|
|
|
|
target = Image.open(TEST_FILE_DXT3.replace(".dds", ".png"))
|
|
|
|
im = Image.open(TEST_FILE_DXT3)
|
|
im.load()
|
|
|
|
self.assertEqual(im.format, "DDS")
|
|
self.assertEqual(im.mode, "RGBA")
|
|
self.assertEqual(im.size, (256, 256))
|
|
|
|
self.assert_image_equal(target, im)
|
|
|
|
def test_dx10_bc7(self):
|
|
"""Check DX10 images can be opened"""
|
|
|
|
target = Image.open(TEST_FILE_DX10_BC7.replace(".dds", ".png"))
|
|
|
|
im = Image.open(TEST_FILE_DX10_BC7)
|
|
im.load()
|
|
|
|
self.assertEqual(im.format, "DDS")
|
|
self.assertEqual(im.mode, "RGBA")
|
|
self.assertEqual(im.size, (256, 256))
|
|
|
|
self.assert_image_equal(target, im)
|
|
|
|
def test_unimplemented_dxgi_format(self):
|
|
self.assertRaises(
|
|
NotImplementedError,
|
|
Image.open,
|
|
"Tests/images/unimplemented_dxgi_format.dds",
|
|
)
|
|
|
|
def test_uncompressed_rgb(self):
|
|
"""Check uncompressed RGB images can be opened"""
|
|
|
|
target = Image.open(TEST_FILE_UNCOMPRESSED_RGB.replace(".dds", ".png"))
|
|
|
|
im = Image.open(TEST_FILE_UNCOMPRESSED_RGB)
|
|
im.load()
|
|
|
|
self.assertEqual(im.format, "DDS")
|
|
self.assertEqual(im.mode, "RGBA")
|
|
self.assertEqual(im.size, (800, 600))
|
|
|
|
self.assert_image_equal(target, im)
|
|
|
|
def test__validate_true(self):
|
|
"""Check valid prefix"""
|
|
# Arrange
|
|
prefix = b"DDS etc"
|
|
|
|
# Act
|
|
output = DdsImagePlugin._validate(prefix)
|
|
|
|
# Assert
|
|
self.assertTrue(output)
|
|
|
|
def test__validate_false(self):
|
|
"""Check invalid prefix"""
|
|
# Arrange
|
|
prefix = b"something invalid"
|
|
|
|
# Act
|
|
output = DdsImagePlugin._validate(prefix)
|
|
|
|
# Assert
|
|
self.assertFalse(output)
|
|
|
|
def test_short_header(self):
|
|
""" Check a short header"""
|
|
with open(TEST_FILE_DXT5, "rb") as f:
|
|
img_file = f.read()
|
|
|
|
def short_header():
|
|
Image.open(BytesIO(img_file[:119]))
|
|
|
|
self.assertRaises(IOError, short_header)
|
|
|
|
def test_short_file(self):
|
|
""" Check that the appropriate error is thrown for a short file"""
|
|
|
|
with open(TEST_FILE_DXT5, "rb") as f:
|
|
img_file = f.read()
|
|
|
|
def short_file():
|
|
im = Image.open(BytesIO(img_file[:-100]))
|
|
im.load()
|
|
|
|
self.assertRaises(IOError, short_file)
|
|
|
|
def test_unimplemented_pixel_format(self):
|
|
self.assertRaises(
|
|
NotImplementedError,
|
|
Image.open,
|
|
"Tests/images/unimplemented_pixel_format.dds",
|
|
)
|