mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-25 01:16:16 +03:00
43 lines
1012 B
Python
43 lines
1012 B
Python
from helper import unittest, PillowTestCase, hopper
|
|
|
|
from PIL import Image, XpmImagePlugin
|
|
|
|
# sample ppm stream
|
|
TEST_FILE = "Tests/images/hopper.xpm"
|
|
|
|
|
|
class TestFileXpm(PillowTestCase):
|
|
|
|
def test_sanity(self):
|
|
im = Image.open(TEST_FILE)
|
|
im.load()
|
|
self.assertEqual(im.mode, "P")
|
|
self.assertEqual(im.size, (128, 128))
|
|
self.assertEqual(im.format, "XPM")
|
|
|
|
# large error due to quantization->44 colors.
|
|
self.assert_image_similar(im.convert('RGB'), hopper('RGB'), 60)
|
|
|
|
def test_invalid_file(self):
|
|
invalid_file = "Tests/images/flower.jpg"
|
|
|
|
self.assertRaises(SyntaxError,
|
|
lambda: XpmImagePlugin.XpmImageFile(invalid_file))
|
|
|
|
def test_load_read(self):
|
|
# Arrange
|
|
im = Image.open(TEST_FILE)
|
|
dummy_bytes = 1
|
|
|
|
# Act
|
|
data = im.load_read(dummy_bytes)
|
|
|
|
# Assert
|
|
self.assertEqual(len(data), 16384)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|
|
# End of file
|