""" Part of the bsie test suite. A copy of the license is provided with the project. Author: Matthias Baumgartner, 2022 """ # standard imports import os import unittest # external imports import PIL.Image # bsie imports from bsie.utils import errors # objects to test from bsie.reader.image._pillow import PillowImage ## code ## class TestPillowImage(unittest.TestCase): def test_call(self): rdr = PillowImage() # returns PIL image img = rdr(os.path.join(os.path.dirname(__file__), 'testimage.jpg')) self.assertEqual(img.size, (1, 1)) self.assertEqual(img.getdata().getpixel((0, 0)), (0, 0, 0)) img.close() # raises exception when image cannot be read self.assertRaises(errors.ReaderError, rdr, os.path.join(os.path.dirname(__file__), 'invalid.jpg')) # NOTE: PIL can actually read raw image files (returns the thumbnail) #self.assertRaises(errors.ReaderError, rdr, # os.path.join(os.path.dirname(__file__), 'testimage.nef')) ## main ## if __name__ == '__main__': unittest.main() ## EOF ##