flake8 and tests

PIL/TgaImagePlugin.py

@@ -19,7 +19,6 @@
 
 __version__ = "0.3"
 
-import os
 from PIL import Image, ImageFile, ImagePalette, _binary
 
 
@@ -77,7 +76,7 @@ class TgaImageFile(ImageFile.ImageFile):
         if imagetype in (3, 11):
             self.mode = "L"
             if depth == 1:
-                self.mode = "1" # ???
+                self.mode = "1"  # ???
         elif imagetype in (1, 9):
             self.mode = "P"
         elif imagetype in (2, 10):
@@ -102,24 +101,24 @@ class TgaImageFile(ImageFile.ImageFile):
             self.info["compression"] = "tga_rle"
 
         if idlen:
-            self.fp.seek(idlen, os.SEEK_CUR)
+            self.info["id_section"] = self.fp.read(idlen)
 
         if colormaptype:
             # read palette
             start, size, mapdepth = i16(s[3:]), i16(s[5:]), i16(s[7:])
             if mapdepth == 16:
-                self.palette = ImagePalette.raw("BGR;16",
-                    b"\0"*2*start + self.fp.read(2*size))
+                self.palette = ImagePalette.raw(
+                    "BGR;16", b"\0"*2*start + self.fp.read(2*size))
             elif mapdepth == 24:
-                self.palette = ImagePalette.raw("BGR",
-                    b"\0"*3*start + self.fp.read(3*size))
+                self.palette = ImagePalette.raw(
+                    "BGR", b"\0"*3*start + self.fp.read(3*size))
             elif mapdepth == 32:
-                self.palette = ImagePalette.raw("BGRA",
-                    b"\0"*4*start + self.fp.read(4*size))
+                self.palette = ImagePalette.raw(
+                    "BGRA", b"\0"*4*start + self.fp.read(4*size))
 
         # setup tile descriptor
         try:
-            rawmode = MODES[(imagetype&7, depth)]
+            rawmode = MODES[(imagetype & 7, depth)]
             if imagetype & 8:
                 # compressed
                 self.tile = [("tga_rle", (0, 0)+self.size,
@@ -128,7 +127,7 @@ class TgaImageFile(ImageFile.ImageFile):
                 self.tile = [("raw", (0, 0)+self.size,
                               self.fp.tell(), (rawmode, 0, orientation))]
         except KeyError:
-            pass # cannot decode
+            pass  # cannot decode
 
 #
 # --------------------------------------------------------------------
@@ -146,6 +145,7 @@ SAVE = {
     "RGBA": ("BGRA", 32, 0, 2),
 }
 
+
 def _save(im, fp, filename, check=0):
 
     try:
@@ -186,7 +186,8 @@ def _save(im, fp, filename, check=0):
     if colormaptype:
         fp.write(im.im.getpalette("RGB", "BGR"))
 
-    ImageFile._save(im, fp, [("raw", (0,0)+im.size, 0, (rawmode, 0, orientation))])
+    ImageFile._save(
+        im, fp, [("raw", (0, 0) + im.size, 0, (rawmode, 0, orientation))])
 
 #
 # --------------------------------------------------------------------

Tests/test_file_tga.py

@@ -0,0 +1,22 @@
+from helper import unittest, PillowTestCase
+
+from PIL import Image
+
+
+class TestFileSun(PillowTestCase):
+
+    def test_sanity(self):
+        # tga file with id field
+        test_file = "Tests/images/tga_id_field.tga"
+
+        # Act
+        im = Image.open(test_file)
+
+        # Assert
+        self.assertEqual(im.size, (100, 100))
+
+
+if __name__ == '__main__':
+    unittest.main()
+
+# End of file