Added test for ImageTk _get_image_from_kw

This commit is contained in:
Andrew Murray 2016-05-01 22:19:52 +10:00
parent d82935a2d9
commit 0b515ae3fa

View File

@ -1,15 +1,43 @@
from helper import unittest, PillowTestCase from helper import unittest, PillowTestCase
from PIL import Image
try:
from PIL import ImageTk
dir(ImageTk)
except (OSError, ImportError) as v:
# Skipped via setUp()
pass
class TestImageTk(PillowTestCase): class TestImageTk(PillowTestCase):
def test_import(self): def setUp(self):
try: try:
from PIL import ImageTk from PIL import ImageTk
dir(ImageTk) dir(ImageTk)
except (OSError, ImportError) as v: except (OSError, ImportError) as v:
self.skipTest(v) self.skipTest(v)
def test_kw(self):
TEST_JPG = "Tests/images/hopper.jpg"
TEST_PNG = "Tests/images/hopper.png"
im1 = Image.open(TEST_JPG)
im2 = Image.open(TEST_PNG)
with open(TEST_PNG, 'rb') as fp:
data = fp.read()
kw = {"file":TEST_JPG, "data":data}
# Test "file"
im = ImageTk._get_image_from_kw(kw)
self.assert_image_equal(im, im1)
# Test "data"
im = ImageTk._get_image_from_kw(kw)
self.assert_image_equal(im, im2)
# Test no relevant entry
im = ImageTk._get_image_from_kw(kw)
self.assertEqual(im, None)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()