test it

Author: Matt Land

adafruit_imageload/tests/test_pnm_load.py

@@ -61,6 +61,7 @@ def __str__(self):
             out += "\n"
         return out
 
+
 class Palette_C_Interface(object):
     def __init__(self, num_colors):
         self.num_colors = num_colors
@@ -143,8 +144,9 @@ def test_decode(self):
         encoded = b._abs_pos(3, 3)
         self.assertEqual((3, 3), b._decode(encoded))
 
-class testPgmLoad(TestCase):
-    def test_load_works_p1_ascii(self):
+
+class TestPgmLoad(TestCase):
+    def test_load_works_p2_ascii(self):
         test_file = os.path.join(
             os.path.dirname(__file__),
             "..",
@@ -160,8 +162,25 @@ def test_load_works_p1_ascii(self):
         self.assertEqual(8, bitmap.width)
         self.assertEqual(8, bitmap.height)
         bitmap.validate()
-        self.fail(str(bitmap))
+        self.assertEqual(6, palette.num_colors)
 
+    def test_load_works_p5_binary(self):
+        test_file = os.path.join(
+            os.path.dirname(__file__),
+            "..",
+            "..",
+            "examples",
+            "images",
+            "netpbm_p5_binary.pgm",
+        )
+        with open(test_file, "rb") as f:
+            bitmap, palette = pnm.load(f, b"P5", bitmap=Bitmap_C_Interface, palette=Palette_C_Interface)
+        self.assertTrue(isinstance(bitmap, Bitmap_C_Interface), bitmap)
+        self.assertEqual(255, bitmap.colors)
+        self.assertEqual(8, bitmap.width)
+        self.assertEqual(8, bitmap.height)
+        bitmap.validate()
+        self.assertEqual(6, palette.colors)
 
 
 class TestPnmLoad(TestCase):