Pillow/Tests/test_file_ppm.py

302 lines
7.9 KiB
Python
Raw Normal View History

2021-08-10 12:56:52 +03:00
import sys
from io import BytesIO
import pytest
2021-02-01 02:47:02 +03:00
from PIL import Image, PpmImagePlugin
from .helper import assert_image_equal_tofile, 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
assert_image_equal_tofile(im, "Tests/images/16_bit_binary_pgm.png")
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
assert_image_equal_tofile(im, f)
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
assert_image_equal_tofile(im, f)
2016-12-28 01:54:10 +03:00
2021-02-01 02:47:02 +03:00
def test_plain_pbm(tmp_path):
2022-03-12 09:32:15 +03:00
# P1
with Image.open("Tests/images/hopper_1bit_plain.pbm") as im:
2022-03-12 09:32:15 +03:00
# P4
assert_image_equal_tofile(im, "Tests/images/hopper_1bit.pbm")
2021-02-01 02:47:02 +03:00
def test_8bit_plain_pgm(tmp_path):
2022-03-12 09:32:15 +03:00
# P2
with Image.open("Tests/images/hopper_8bit_plain.pgm") as im:
2022-03-12 09:32:15 +03:00
# P5
assert_image_equal_tofile(im, "Tests/images/hopper_8bit.pgm")
2021-02-01 02:47:02 +03:00
def test_8bit_plain_ppm(tmp_path):
2022-03-12 09:32:15 +03:00
# P3
with Image.open("Tests/images/hopper_8bit_plain.ppm") as im:
2022-03-12 09:32:15 +03:00
# P6
assert_image_equal_tofile(im, "Tests/images/hopper_8bit.ppm")
2021-02-01 02:47:02 +03:00
def test_16bit_plain_pgm(tmp_path):
2022-03-12 09:32:15 +03:00
# P2 with maxval 2 ** 16 - 1
with Image.open("Tests/images/hopper_16bit_plain.pgm") as im:
assert im.mode == "I"
assert im.size == (128, 128)
assert im.get_format_mimetype() == "image/x-portable-graymap"
2021-02-01 02:47:02 +03:00
2022-03-12 09:32:15 +03:00
# P5 with maxval 2 ** 16 - 1
assert_image_equal_tofile(im, "Tests/images/hopper_16bit.pgm")
2021-02-01 02:47:02 +03:00
def test_32bit_plain_pgm(tmp_path):
2022-03-12 09:32:15 +03:00
# P2 with maxval 2 ** 31 - 1
with Image.open("Tests/images/hopper_32bit_plain.pgm") as im:
assert im.mode == "I"
assert im.size == (128, 128)
assert im.get_format_mimetype() == "image/x-portable-graymap"
2021-02-01 02:47:02 +03:00
2022-03-12 09:32:15 +03:00
# P5 with maxval 2 ** 31 - 1
assert_image_equal_tofile(im, "Tests/images/hopper_32bit.pgm")
2021-02-01 02:47:02 +03:00
def test_plain_pbm_data_with_comments(tmp_path):
path1 = str(tmp_path / "temp1.ppm")
path2 = str(tmp_path / "temp2.ppm")
comment = b"# veeery long comment" * 10**6
2021-02-01 02:47:02 +03:00
with open(path1, "wb") as f1, open(path2, "wb") as f2:
f1.write(b"P1\n2 2\n\n1010")
f2.write(b"P1\n2 2\n" + comment + b"\n1010" + comment)
with Image.open(path1) as im:
assert_image_equal_tofile(im, path2)
2021-02-01 02:47:02 +03:00
def test_plain_pbm_truncated_data(tmp_path):
2021-03-21 08:16:39 +03:00
path = str(tmp_path / "temp.ppm")
with open(path, "wb") as f:
2021-02-01 02:47:02 +03:00
f.write(b"P1\n128 128\n")
2022-03-04 09:08:10 +03:00
with Image.open(path) as im:
with pytest.raises(ValueError):
im.load()
2021-02-01 02:47:02 +03:00
def test_plain_pbm_invalid_data(tmp_path):
path = str(tmp_path / "temp.ppm")
with open(path, "wb") as f:
f.write(b"P1\n128 128\n1009")
2022-03-04 09:08:10 +03:00
with Image.open(path) as im:
with pytest.raises(ValueError):
im.load()
2021-02-01 02:47:02 +03:00
def test_plain_ppm_data_with_comments(tmp_path):
path1 = str(tmp_path / "temp1.ppm")
path2 = str(tmp_path / "temp2.ppm")
comment = b"# veeery long comment" * 10**6
2021-02-01 02:47:02 +03:00
with open(path1, "wb") as f1, open(path2, "wb") as f2:
f1.write(b"P3\n2 2\n255\n0 0 0 001 1 1 2 2 2 255 255 255")
f2.write(
b"P3\n2 2\n255\n" + comment + b"\n0 0 0 001 1 1 2 2 2 255 255 255" + comment
)
with Image.open(path1) as im:
assert_image_equal_tofile(im, path2)
2021-02-01 02:47:02 +03:00
def test_plain_ppm_truncated_data(tmp_path):
path = str(tmp_path / "temp.ppm")
with open(path, "wb") as f:
f.write(b"P3\n128 128\n255\n")
2022-03-04 09:08:10 +03:00
with Image.open(path) as im:
with pytest.raises(ValueError):
im.load()
2021-02-01 02:47:02 +03:00
def test_plain_ppm_invalid_data(tmp_path):
path = str(tmp_path / "temp.ppm")
with open(path, "wb") as f:
f.write(b"P3\n128 128\n255\n100A")
2022-03-04 09:08:10 +03:00
with Image.open(path) as im:
with pytest.raises(ValueError):
im.load()
2021-02-01 02:47:02 +03:00
def test_plain_ppm_half_token_too_long(tmp_path):
path = str(tmp_path / "temp.ppm")
with open(path, "wb") as f:
f.write(b"P3\n128 128\n255\n012345678910")
2022-03-04 09:08:10 +03:00
with Image.open(path) as im:
with pytest.raises(ValueError):
im.load()
2021-02-01 02:47:02 +03:00
def test_plain_ppm_token_too_long(tmp_path):
path = str(tmp_path / "temp.ppm")
with open(path, "wb") as f:
f.write(b"P3\n128 128\n255\n012345678910 0")
2022-03-04 09:08:10 +03:00
with Image.open(path) as im:
with pytest.raises(ValueError):
im.load()
2021-02-01 02:47:02 +03:00
def test_plain_ppm_value_too_large(tmp_path):
path = str(tmp_path / "temp.ppm")
with open(path, "wb") as f:
f.write(b"P3\n128 128\n255\n256")
2022-03-04 09:08:10 +03:00
with Image.open(path) as im:
with pytest.raises(ValueError):
im.load()
2021-02-01 02:47:02 +03:00
def test_magic(tmp_path):
with pytest.raises(SyntaxError):
PpmImagePlugin.PpmImageFile(fp=BytesIO(b"PyInvalid"))
2019-03-04 10:17:12 +03:00
2020-12-21 07:15:49 +03:00
def test_header_with_comments(tmp_path):
path = str(tmp_path / "temp.ppm")
with open(path, "wb") as f:
2022-03-12 09:32:15 +03:00
f.write(b"P6 #comment\n#comment\r12#comment\r8\n128 #comment\n255\n")
2020-12-21 07:15:49 +03:00
with Image.open(path) as im:
assert im.size == (128, 128)
2022-03-04 07:22:41 +03:00
def test_non_integer_token(tmp_path):
2021-03-21 08:16:39 +03:00
path = str(tmp_path / "temp.ppm")
2020-12-21 07:15:49 +03:00
with open(path, "wb") as f:
2022-03-04 07:22:41 +03:00
f.write(b"P6\nTEST")
2020-12-21 07:15:49 +03:00
2021-01-06 07:15:07 +03:00
with pytest.raises(ValueError):
2021-03-21 06:42:36 +03:00
with Image.open(path):
pass
2020-12-21 07:15:49 +03:00
2021-02-01 02:47:02 +03:00
def test_header_token_too_long(tmp_path):
2021-03-21 08:16:39 +03:00
path = str(tmp_path / "temp.ppm")
with open(path, "wb") as f:
2022-03-04 07:22:41 +03:00
f.write(b"P6\n 01234567890")
2022-03-04 07:22:41 +03:00
with pytest.raises(ValueError) as e:
2021-03-21 06:42:36 +03:00
with Image.open(path):
pass
2020-12-21 07:15:49 +03:00
2022-03-04 07:22:41 +03:00
assert str(e.value) == "Token too long in file header: b'01234567890'"
def test_too_many_colors(tmp_path):
2021-03-21 08:16:39 +03:00
path = str(tmp_path / "temp.ppm")
with open(path, "wb") as f:
f.write(b"P6\n1 1\n1000\n")
2022-03-04 07:22:41 +03:00
with pytest.raises(ValueError) as e:
2021-03-21 06:42:36 +03:00
with Image.open(path):
pass
2022-03-04 07:22:41 +03:00
assert str(e.value) == "Too many colors for band: 1000"
2021-02-01 02:47:02 +03:00
def test_truncated_header(tmp_path):
2020-02-25 12:57:27 +03:00
path = str(tmp_path / "temp.pgm")
with open(path, "w") as f:
f.write("P6")
2019-03-04 10:17:12 +03:00
2022-03-04 07:22:41 +03:00
with pytest.raises(ValueError) as e:
2021-02-11 13:43:54 +03:00
with Image.open(path):
pass
2019-03-04 10:17:12 +03:00
2022-03-04 07:22:41 +03:00
assert str(e.value) == "Reached EOF while reading header"
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.
2021-01-06 07:15:07 +03:00
with pytest.raises(OSError):
2021-02-11 13:43:54 +03:00
with Image.open("Tests/images/negative_size.ppm"):
pass
2020-02-25 12:57:27 +03:00
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"
2021-08-10 12:56:52 +03:00
@pytest.mark.parametrize("buffer", (True, False))
def test_save_stdout(buffer):
old_stdout = sys.stdout
if buffer:
class MyStdOut:
buffer = BytesIO()
mystdout = MyStdOut()
else:
mystdout = BytesIO()
sys.stdout = mystdout
with Image.open(TEST_FILE) as im:
im.save(sys.stdout, "PPM")
# Reset stdout
sys.stdout = old_stdout
if buffer:
mystdout = mystdout.buffer
2021-08-24 16:43:38 +03:00
with Image.open(mystdout) as reloaded:
assert_image_equal_tofile(reloaded, TEST_FILE)