mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-21 03:18:57 +03:00
61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
from helper import unittest, PillowTestCase
|
|
|
|
from PIL import Image
|
|
|
|
# sample ppm stream
|
|
test_file = "Tests/images/hopper.ppm"
|
|
data = open(test_file, "rb").read()
|
|
|
|
|
|
class TestFilePpm(PillowTestCase):
|
|
|
|
def test_sanity(self):
|
|
im = Image.open(test_file)
|
|
im.load()
|
|
self.assertEqual(im.mode, "RGB")
|
|
self.assertEqual(im.size, (128, 128))
|
|
self.assertEqual(im.format, "PPM")
|
|
|
|
def test_16bit_pgm(self):
|
|
im = Image.open('Tests/images/16_bit_binary.pgm')
|
|
im.load()
|
|
self.assertEqual(im.mode, 'I')
|
|
self.assertEqual(im.size, (20, 100))
|
|
|
|
tgt = Image.open('Tests/images/16_bit_binary_pgm.png')
|
|
self.assert_image_equal(im, tgt)
|
|
|
|
def test_16bit_pgm_write(self):
|
|
im = Image.open('Tests/images/16_bit_binary.pgm')
|
|
im.load()
|
|
|
|
f = self.tempfile('temp.pgm')
|
|
im.save(f, 'PPM')
|
|
|
|
reloaded = Image.open(f)
|
|
self.assert_image_equal(im, reloaded)
|
|
|
|
def test_truncated_file(self):
|
|
path = self.tempfile('temp.pgm')
|
|
f = open(path, 'w')
|
|
f.write('P6')
|
|
f.close()
|
|
|
|
self.assertRaises(ValueError, lambda: Image.open(path))
|
|
|
|
|
|
def test_neg_ppm(self):
|
|
"""test_neg_ppm
|
|
|
|
Storage.c accepted negative values for xsize, ysize.
|
|
open_ppm is a core debugging item that doesn't check any parameters for
|
|
sanity.
|
|
"""
|
|
|
|
with self.assertRaises(ValueError):
|
|
Image.core.open_ppm('Tests/images/negative_size.ppm')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|