Pillow/Tests/test_file_sgi.py

96 lines
2.8 KiB
Python
Raw Normal View History

from helper import unittest, PillowTestCase, hopper
2014-07-16 22:28:17 +04:00
2015-07-03 08:03:25 +03:00
from PIL import Image, SgiImagePlugin
2014-07-16 22:28:17 +04:00
class TestFileSgi(PillowTestCase):
def test_rgb(self):
# Created with ImageMagick then renamed:
# convert hopper.ppm -compress None sgi:hopper.rgb
test_file = "Tests/images/hopper.rgb"
2014-07-16 22:28:17 +04:00
im = Image.open(test_file)
self.assert_image_equal(im, hopper())
self.assertEqual(im.get_format_mimetype(), 'image/rgb')
2014-07-16 22:28:17 +04:00
2017-07-29 00:12:46 +03:00
def test_rgb16(self):
test_file = "Tests/images/hopper16.rgb"
im = Image.open(test_file)
self.assert_image_equal(im, hopper())
2014-07-16 22:28:17 +04:00
def test_l(self):
# Created with ImageMagick
# convert hopper.ppm -monochrome -compress None sgi:hopper.bw
test_file = "Tests/images/hopper.bw"
2014-07-16 22:28:17 +04:00
im = Image.open(test_file)
self.assert_image_similar(im, hopper('L'), 2)
self.assertEqual(im.get_format_mimetype(), 'image/sgi')
2014-07-16 22:28:17 +04:00
def test_rgba(self):
# Created with ImageMagick:
# convert transparent.png -compress None transparent.sgi
2014-07-16 22:28:17 +04:00
test_file = "Tests/images/transparent.sgi"
2017-04-20 14:14:23 +03:00
im = Image.open(test_file)
target = Image.open('Tests/images/transparent.png')
self.assert_image_equal(im, target)
self.assertEqual(im.get_format_mimetype(), 'image/sgi')
def test_rle(self):
2017-07-22 10:34:06 +03:00
# Created with ImageMagick:
# convert hopper.ppm hopper.sgi
test_file = "Tests/images/hopper.sgi"
2017-07-22 10:34:06 +03:00
im = Image.open(test_file)
target = Image.open('Tests/images/hopper.rgb')
self.assert_image_equal(im, target)
2017-07-29 00:12:46 +03:00
def test_rle16(self):
test_file = "Tests/images/tv16.sgi"
im = Image.open(test_file)
target = Image.open('Tests/images/tv.rgb')
self.assert_image_equal(im, target)
2014-07-16 22:28:17 +04:00
2015-07-03 08:03:25 +03:00
def test_invalid_file(self):
2015-07-03 09:22:56 +03:00
invalid_file = "Tests/images/flower.jpg"
self.assertRaises(ValueError,
2017-09-01 14:05:40 +03:00
SgiImagePlugin.SgiImageFile, invalid_file)
2015-07-03 08:03:25 +03:00
2016-12-31 01:31:35 +03:00
def test_write(self):
def roundtrip(img):
out = self.tempfile('temp.sgi')
img.save(out, format='sgi')
reloaded = Image.open(out)
self.assert_image_equal(img, reloaded)
for mode in ('L', 'RGB', 'RGBA'):
roundtrip(hopper(mode))
# Test 1 dimension for an L mode image
roundtrip(Image.new('L', (10, 1)))
2017-09-29 13:39:43 +03:00
def test_write16(self):
test_file = "Tests/images/hopper16.rgb"
im = Image.open(test_file)
out = self.tempfile('temp.sgi')
im.save(out, format='sgi', bpc=2)
reloaded = Image.open(out)
self.assert_image_equal(im, reloaded)
2018-01-27 09:07:24 +03:00
def test_unsupported_mode(self):
im = hopper('LA')
out = self.tempfile('temp.sgi')
2017-09-01 14:05:40 +03:00
self.assertRaises(ValueError, im.save, out, format='sgi')
2016-12-31 01:31:35 +03:00
2014-07-16 22:28:17 +04:00
if __name__ == '__main__':
unittest.main()