Pillow/Tests/test_file_bmp.py

140 lines
3.8 KiB
Python
Raw Normal View History

2014-01-23 08:40:37 +04:00
import io
2020-02-17 11:42:33 +03:00
import pytest
from PIL import BmpImagePlugin, Image
2020-02-17 11:42:33 +03:00
from .helper import assert_image_equal, hopper
2020-02-17 11:42:33 +03:00
def test_sanity(tmp_path):
def roundtrip(im):
outfile = str(tmp_path / "temp.bmp")
2019-06-13 18:53:42 +03:00
im.save(outfile, "BMP")
2019-11-25 23:03:23 +03:00
with Image.open(outfile) as reloaded:
reloaded.load()
2020-02-17 11:42:33 +03:00
assert im.mode == reloaded.mode
assert im.size == reloaded.size
assert reloaded.format == "BMP"
assert reloaded.get_format_mimetype() == "image/bmp"
roundtrip(hopper())
roundtrip(hopper("1"))
roundtrip(hopper("L"))
roundtrip(hopper("P"))
roundtrip(hopper("RGB"))
2019-03-30 07:03:57 +03:00
2020-02-17 11:42:33 +03:00
def test_invalid_file():
with open("Tests/images/flower.jpg", "rb") as fp:
with pytest.raises(SyntaxError):
BmpImagePlugin.BmpImageFile(fp)
2020-02-17 11:42:33 +03:00
def test_save_to_bytes():
output = io.BytesIO()
im = hopper()
im.save(output, "BMP")
2020-02-17 11:42:33 +03:00
output.seek(0)
with Image.open(output) as reloaded:
assert im.mode == reloaded.mode
assert im.size == reloaded.size
assert reloaded.format == "BMP"
def test_save_too_large(tmp_path):
outfile = str(tmp_path / "temp.bmp")
with Image.new("RGB", (1, 1)) as im:
im._size = (37838, 37838)
with pytest.raises(ValueError):
2019-11-25 23:03:23 +03:00
im.save(outfile)
2020-02-17 11:42:33 +03:00
def test_dpi():
dpi = (72, 72)
output = io.BytesIO()
with hopper() as im:
im.save(output, "BMP", dpi=dpi)
output.seek(0)
with Image.open(output) as reloaded:
assert reloaded.info["dpi"] == dpi
def test_save_bmp_with_dpi(tmp_path):
# Test for #1301
# Arrange
outfile = str(tmp_path / "temp.jpg")
with Image.open("Tests/images/hopper.bmp") as im:
# Act
im.save(outfile, "JPEG", dpi=im.info["dpi"])
# Assert
with Image.open(outfile) as reloaded:
reloaded.load()
assert im.info["dpi"] == reloaded.info["dpi"]
assert im.size == reloaded.size
assert reloaded.format == "JPEG"
def test_load_dpi_rounding():
# Round up
with Image.open("Tests/images/hopper.bmp") as im:
assert im.info["dpi"] == (96, 96)
2020-02-17 11:42:33 +03:00
# Round down
with Image.open("Tests/images/hopper_roundDown.bmp") as im:
assert im.info["dpi"] == (72, 72)
2020-02-17 11:42:33 +03:00
def test_save_dpi_rounding(tmp_path):
outfile = str(tmp_path / "temp.bmp")
with Image.open("Tests/images/hopper.bmp") as im:
im.save(outfile, dpi=(72.2, 72.2))
with Image.open(outfile) as reloaded:
assert reloaded.info["dpi"] == (72, 72)
im.save(outfile, dpi=(72.8, 72.8))
with Image.open(outfile) as reloaded:
assert reloaded.info["dpi"] == (73, 73)
def test_load_dib():
# test for #1293, Imagegrab returning Unsupported Bitfields Format
with Image.open("Tests/images/clipboard.dib") as im:
assert im.format == "DIB"
assert im.get_format_mimetype() == "image/bmp"
with Image.open("Tests/images/clipboard_target.png") as target:
assert_image_equal(im, target)
2020-02-17 11:42:33 +03:00
def test_save_dib(tmp_path):
outfile = str(tmp_path / "temp.dib")
with Image.open("Tests/images/clipboard.dib") as im:
im.save(outfile)
with Image.open(outfile) as reloaded:
assert reloaded.format == "DIB"
assert reloaded.get_format_mimetype() == "image/bmp"
assert_image_equal(im, reloaded)
def test_rgba_bitfields():
# This test image has been manually hexedited
# to change the bitfield compression in the header from XBGR to RGBA
with Image.open("Tests/images/rgb32bf-rgba.bmp") as im:
# So before the comparing the image, swap the channels
b, g, r = im.split()[1:]
im = Image.merge("RGB", (r, g, b))
with Image.open("Tests/images/bmp/q/rgb32bf-xbgr.bmp") as target:
assert_image_equal(im, target)