from helper import unittest, PillowTestCase, hopper from PIL import Image, SgiImagePlugin 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" im = Image.open(test_file) self.assert_image_equal(im, hopper()) def test_rgb16(self): test_file = "Tests/images/hopper16.rgb" im = Image.open(test_file) self.assert_image_equal(im, hopper()) def test_l(self): # Created with ImageMagick # convert hopper.ppm -monochrome -compress None sgi:hopper.bw test_file = "Tests/images/hopper.bw" im = Image.open(test_file) self.assert_image_similar(im, hopper('L'), 2) def test_rgba(self): # Created with ImageMagick: # convert transparent.png -compress None transparent.sgi test_file = "Tests/images/transparent.sgi" im = Image.open(test_file) target = Image.open('Tests/images/transparent.png') self.assert_image_equal(im, target) def test_rle(self): # Created with ImageMagick: # convert hopper.ppm hopper.sgi test_file = "Tests/images/hopper.sgi" im = Image.open(test_file) target = Image.open('Tests/images/hopper.rgb') self.assert_image_equal(im, target) 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) def test_invalid_file(self): invalid_file = "Tests/images/flower.jpg" self.assertRaises(ValueError, SgiImagePlugin.SgiImageFile, invalid_file) 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))) 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) def test_unsupported_mode(self): im = hopper('LA') out = self.tempfile('temp.sgi') self.assertRaises(ValueError, im.save, out, format='sgi') if __name__ == '__main__': unittest.main()