addtests

Author: Matt Land

adafruit_imageload/tests/__init__.py

@@ -0,0 +1,56 @@
+from unittest import TestCase
+from adafruit_imageload import pnm
+from io import BytesIO
+import logging
+import os
+
+class Bitmap(object):
+    def __init__(self, width, height, colors):
+        self.width = width
+        self.height = height
+        self.colors = colors
+
+logging.getLogger().setLevel(logging.INFO)
+
+class TestPbmP1(TestCase):
+
+    def test_load_fails_with_no_header_data(self):
+        f = BytesIO(b"some initial binary data: \x00\x01")
+        try:
+            pnm.load(f, b'P1')
+            self.fail('should have failed')
+        except Exception as e:
+            if "Unsupported image format" not in str(e):
+                raise
+
+    def _test_load_works_p1_ascii(self):
+        test_file = os.path.join(os.path.dirname(__file__), '..', '..', 'examples', 'images', 'netpbm_p1_mono.pbm')
+        with open(test_file) as f:
+            f.seek(2)
+            bitmap, palette = pnm.load(f, b'P1')
+        self.assertTrue(isinstance(bitmap, Bitmap))
+        self.fail(bitmap)
+        self.assertEqual(1, bitmap.colors)
+        self.assertEqual(20, bitmap.width)
+
+    def test_load_works_p4_binary(self):
+        test_file = os.path.join(os.path.dirname(__file__), '..', '..', 'examples', 'images', 'netpbm_p4_mono.pbm')
+
+        with open(test_file, 'rb') as f:
+            f.seek(2)
+            bitmap, palette = pnm.load(f, b'P4', bitmap=Bitmap)
+        self.assertTrue(isinstance(bitmap, Bitmap))
+        self.assertEqual(1, bitmap.colors)
+        self.assertEqual(8, bitmap.width)
+        self.assertEqual(8, bitmap.height)
+
+    def test_load_works_p4_binary_big(self):
+        test_file = os.path.join(os.path.dirname(__file__), '..', '..', 'examples', 'images', 'MARBLES.PBM')
+
+        with open(test_file, 'rb') as f:
+            f.seek(2)
+            bitmap, palette = pnm.load(f, b'P4', bitmap=Bitmap)
+        self.assertTrue(isinstance(bitmap, Bitmap))
+        self.assertEqual(1, bitmap.colors)
+        self.assertEqual(1152, bitmap.width)
+        self.assertEqual(813, bitmap.height)