Merge 5f8e559521 into 8e6ef35f78

PIL/PalmImagePlugin.py

@@ -77,6 +77,7 @@ _Palm8BitColormapValues = (
     (  0,   0,   0), (  0,   0,   0), (  0,   0,   0), (  0,   0,   0),
     (  0,   0,   0), (  0,   0,   0), (  0,   0,   0), (  0,   0,   0))
 
+
 # so build a prototype image to be used for palette resampling
 def build_prototype_image():
     image = Image.new("L", (1, len(_Palm8BitColormapValues),))
@@ -91,7 +92,8 @@ def build_prototype_image():
 
 Palm8BitColormapImage = build_prototype_image()
 
-# OK, we now have in Palm8BitColormapImage, a "P"-mode image with the right palette
+# OK, we now have in Palm8BitColormapImage,
+# a "P"-mode image with the right palette
 #
 # --------------------------------------------------------------------
 
@@ -110,6 +112,7 @@ _COMPRESSION_TYPES = {
 o8 = _binary.o8
 o16b = _binary.o16be
 
+
 #
 # --------------------------------------------------------------------
 
@@ -127,12 +130,16 @@ def _save(im, fp, filename, check=0):
         bpp = 8
         version = 1
 
-    elif im.mode == "L" and "bpp" in im.encoderinfo and im.encoderinfo["bpp"] in (1, 2, 4):
+    elif (im.mode == "L" and
+          "bpp" in im.encoderinfo and
+          im.encoderinfo["bpp"] in (1, 2, 4)):
 
-        # this is 8-bit grayscale, so we shift it to get the high-order bits, and invert it because
+        # this is 8-bit grayscale, so we shift it to get the high-order bits,
+        # and invert it because
         # Palm does greyscale from white (0) to black (1)
         bpp = im.encoderinfo["bpp"]
-        im = im.point(lambda x, shift=8-bpp, maxval=(1 << bpp)-1: maxval - (x >> shift))
+        im = im.point(
+            lambda x, shift=8-bpp, maxval=(1 << bpp)-1: maxval - (x >> shift))
         # we ignore the palette here
         im.mode = "P"
         rawmode = "P;" + str(bpp)
@@ -140,8 +147,9 @@ def _save(im, fp, filename, check=0):
 
     elif im.mode == "L" and "bpp" in im.info and im.info["bpp"] in (1, 2, 4):
 
-        # here we assume that even though the inherent mode is 8-bit grayscale, only
-        # the lower bpp bits are significant.  We invert them to match the Palm.
+        # here we assume that even though the inherent mode is 8-bit grayscale,
+        # only the lower bpp bits are significant.
+        # We invert them to match the Palm.
         bpp = im.info["bpp"]
         im = im.point(lambda x, maxval=(1 << bpp)-1: maxval - (x & maxval))
         # we ignore the palette here
@@ -172,7 +180,7 @@ def _save(im, fp, filename, check=0):
     cols = im.size[0]
     rows = im.size[1]
 
-    rowbytes = ((cols + (16//bpp - 1)) / (16 // bpp)) * 2
+    rowbytes = int((cols + (16//bpp - 1)) / (16 // bpp)) * 2
     transparent_index = 0
     compression_type = _COMPRESSION_TYPES["none"]
 
@@ -205,12 +213,19 @@ def _save(im, fp, filename, check=0):
         for i in range(256):
             fp.write(o8(i))
             if colormapmode == 'RGB':
-                fp.write(o8(colormap[3 * i]) + o8(colormap[3 * i + 1]) + o8(colormap[3 * i + 2]))
+                fp.write(
+                    o8(colormap[3 * i]) +
+                    o8(colormap[3 * i + 1]) +
+                    o8(colormap[3 * i + 2]))
             elif colormapmode == 'RGBA':
-                fp.write(o8(colormap[4 * i]) + o8(colormap[4 * i + 1]) + o8(colormap[4 * i + 2]))
+                fp.write(
+                    o8(colormap[4 * i]) +
+                    o8(colormap[4 * i + 1]) +
+                    o8(colormap[4 * i + 2]))
 
     # now convert data to raw form
-    ImageFile._save(im, fp, [("raw", (0,0)+im.size, 0, (rawmode, rowbytes, 1))])
+    ImageFile._save(
+        im, fp, [("raw", (0, 0)+im.size, 0, (rawmode, rowbytes, 1))])
 
     fp.flush()
 

Tests/test_file_palm.py

@@ -0,0 +1,45 @@
+from helper import unittest, PillowTestCase, tearDownModule, lena
+
+import os.path
+
+
+class TestFilePalm(PillowTestCase):
+
+    def helper_save_as_palm(self, mode):
+        # Arrange
+        im = lena(mode)
+        outfile = self.tempfile("temp_" + mode + ".palm")
+
+        # Act
+        im.save(outfile)
+
+        # Assert
+        self.assertTrue(os.path.isfile(outfile))
+        self.assertGreater(os.path.getsize(outfile), 0)
+
+    def test_monochrome(self):
+        # Arrange
+        mode = "1"
+
+        # Act / Assert
+        self.helper_save_as_palm(mode)
+
+    def test_p_mode(self):
+        # Arrange
+        mode = "P"
+
+        # Act / Assert
+        self.helper_save_as_palm(mode)
+
+    def test_rgb_ioerror(self):
+        # Arrange
+        mode = "RGB"
+
+        # Act / Assert
+        self.assertRaises(IOError, lambda: self.helper_save_as_palm(mode))
+
+
+if __name__ == '__main__':
+    unittest.main()
+
+# End of file