mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-25 17:36:18 +03:00
76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
from PIL import Image
|
|
|
|
from .helper import PillowTestCase, assert_image_equal, assert_image_similar, hopper
|
|
|
|
# sample ppm stream
|
|
test_file = "Tests/images/hopper.ppm"
|
|
|
|
|
|
class TestFilePpm(PillowTestCase):
|
|
def test_sanity(self):
|
|
with Image.open(test_file) as im:
|
|
im.load()
|
|
self.assertEqual(im.mode, "RGB")
|
|
self.assertEqual(im.size, (128, 128))
|
|
self.assertEqual(im.format, "PPM")
|
|
self.assertEqual(im.get_format_mimetype(), "image/x-portable-pixmap")
|
|
|
|
def test_16bit_pgm(self):
|
|
with Image.open("Tests/images/16_bit_binary.pgm") as im:
|
|
im.load()
|
|
self.assertEqual(im.mode, "I")
|
|
self.assertEqual(im.size, (20, 100))
|
|
self.assertEqual(im.get_format_mimetype(), "image/x-portable-graymap")
|
|
|
|
with Image.open("Tests/images/16_bit_binary_pgm.png") as tgt:
|
|
assert_image_equal(im, tgt)
|
|
|
|
def test_16bit_pgm_write(self):
|
|
with Image.open("Tests/images/16_bit_binary.pgm") as im:
|
|
im.load()
|
|
|
|
f = self.tempfile("temp.pgm")
|
|
im.save(f, "PPM")
|
|
|
|
with Image.open(f) as reloaded:
|
|
assert_image_equal(im, reloaded)
|
|
|
|
def test_pnm(self):
|
|
with Image.open("Tests/images/hopper.pnm") as im:
|
|
assert_image_similar(im, hopper(), 0.0001)
|
|
|
|
f = self.tempfile("temp.pnm")
|
|
im.save(f)
|
|
|
|
with Image.open(f) as reloaded:
|
|
assert_image_equal(im, reloaded)
|
|
|
|
def test_truncated_file(self):
|
|
path = self.tempfile("temp.pgm")
|
|
with open(path, "w") as f:
|
|
f.write("P6")
|
|
|
|
self.assertRaises(ValueError, Image.open, path)
|
|
|
|
def test_neg_ppm(self):
|
|
# Storage.c accepted negative values for xsize, ysize. the
|
|
# internal open_ppm function didn't check for sanity but it
|
|
# has been removed. The default opener doesn't accept negative
|
|
# sizes.
|
|
|
|
with self.assertRaises(IOError):
|
|
Image.open("Tests/images/negative_size.ppm")
|
|
|
|
def test_mimetypes(self):
|
|
path = self.tempfile("temp.pgm")
|
|
|
|
with open(path, "w") as f:
|
|
f.write("P4\n128 128\n255")
|
|
with Image.open(path) as im:
|
|
self.assertEqual(im.get_format_mimetype(), "image/x-portable-bitmap")
|
|
|
|
with open(path, "w") as f:
|
|
f.write("PyCMYK\n128 128\n255")
|
|
with Image.open(path) as im:
|
|
self.assertEqual(im.get_format_mimetype(), "image/x-portable-anymap")
|