Pillow/Tests/test_file_ppm.py

83 lines
2.2 KiB
Python
Raw Normal View History

import pytest
from PIL import Image
2020-02-25 12:57:27 +03:00
from .helper import assert_image_equal, assert_image_similar, hopper
# sample ppm stream
2020-02-25 12:57:27 +03:00
TEST_FILE = "Tests/images/hopper.ppm"
2020-02-25 12:57:27 +03:00
def test_sanity():
with Image.open(TEST_FILE) as im:
im.load()
assert im.mode == "RGB"
assert im.size == (128, 128)
assert im.format, "PPM"
assert im.get_format_mimetype() == "image/x-portable-pixmap"
2014-04-08 09:22:42 +04:00
2020-02-25 12:57:27 +03:00
def test_16bit_pgm():
with Image.open("Tests/images/16_bit_binary.pgm") as im:
im.load()
assert im.mode == "I"
assert im.size == (20, 100)
assert im.get_format_mimetype() == "image/x-portable-graymap"
2014-04-08 09:22:42 +04:00
2020-02-25 12:57:27 +03:00
with Image.open("Tests/images/16_bit_binary_pgm.png") as tgt:
assert_image_equal(im, tgt)
2014-04-08 09:22:42 +04:00
2020-02-25 12:57:27 +03:00
def test_16bit_pgm_write(tmp_path):
with Image.open("Tests/images/16_bit_binary.pgm") as im:
im.load()
2014-04-08 09:22:42 +04:00
2020-02-25 12:57:27 +03:00
f = str(tmp_path / "temp.pgm")
im.save(f, "PPM")
2019-03-04 07:11:21 +03:00
2020-02-25 12:57:27 +03:00
with Image.open(f) as reloaded:
assert_image_equal(im, reloaded)
2019-03-04 07:11:21 +03:00
2020-02-25 12:57:27 +03:00
def test_pnm(tmp_path):
with Image.open("Tests/images/hopper.pnm") as im:
assert_image_similar(im, hopper(), 0.0001)
2015-08-28 19:05:08 +03:00
2020-02-25 12:57:27 +03:00
f = str(tmp_path / "temp.pnm")
im.save(f)
2015-08-28 19:05:08 +03:00
2020-02-25 12:57:27 +03:00
with Image.open(f) as reloaded:
assert_image_equal(im, reloaded)
2016-12-28 01:54:10 +03:00
2019-03-04 10:17:12 +03:00
2020-02-25 12:57:27 +03:00
def test_truncated_file(tmp_path):
path = str(tmp_path / "temp.pgm")
with open(path, "w") as f:
f.write("P6")
2019-03-04 10:17:12 +03:00
2020-02-25 12:57:27 +03:00
with pytest.raises(ValueError):
Image.open(path)
2019-03-04 10:17:12 +03:00
2020-02-25 12:57:27 +03:00
def test_neg_ppm():
# 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 pytest.raises(IOError):
Image.open("Tests/images/negative_size.ppm")
def test_mimetypes(tmp_path):
path = str(tmp_path / "temp.pgm")
with open(path, "w") as f:
f.write("P4\n128 128\n255")
with Image.open(path) as im:
assert 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:
assert im.get_format_mimetype() == "image/x-portable-anymap"