mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-10 19:56:47 +03:00
123 lines
3.4 KiB
Python
123 lines
3.4 KiB
Python
import os
|
|
|
|
from PIL import FontFile, Image, ImageDraw, ImageFont, PcfFontFile
|
|
|
|
from .helper import (
|
|
assert_image_equal_tofile,
|
|
assert_image_similar_tofile,
|
|
skip_unless_feature,
|
|
)
|
|
|
|
fontname = "Tests/fonts/ter-x20b.pcf"
|
|
|
|
charsets = {
|
|
"iso8859-1": {
|
|
"glyph_count": 223,
|
|
"message": "hello, world",
|
|
"image1": "Tests/images/test_draw_pbm_ter_en_target.png",
|
|
},
|
|
"iso8859-2": {
|
|
"glyph_count": 223,
|
|
"message": "witaj świecie",
|
|
"image1": "Tests/images/test_draw_pbm_ter_pl_target.png",
|
|
},
|
|
"cp1250": {
|
|
"glyph_count": 250,
|
|
"message": "witaj świecie",
|
|
"image1": "Tests/images/test_draw_pbm_ter_pl_target.png",
|
|
},
|
|
}
|
|
|
|
|
|
pytestmark = skip_unless_feature("zlib")
|
|
|
|
|
|
def save_font(request, tmp_path, encoding):
|
|
with open(fontname, "rb") as test_file:
|
|
font = PcfFontFile.PcfFontFile(test_file, encoding)
|
|
assert isinstance(font, FontFile.FontFile)
|
|
# check the number of characters in the font
|
|
assert len([_f for _f in font.glyph if _f]) == charsets[encoding]["glyph_count"]
|
|
|
|
tempname = str(tmp_path / "temp.pil")
|
|
|
|
def delete_tempfile():
|
|
try:
|
|
os.remove(tempname[:-4] + ".pbm")
|
|
except OSError:
|
|
pass # report?
|
|
|
|
request.addfinalizer(delete_tempfile)
|
|
font.save(tempname)
|
|
|
|
with Image.open(tempname.replace(".pil", ".pbm")) as loaded:
|
|
assert_image_equal_tofile(loaded, f"Tests/fonts/ter-x20b-{encoding}.pbm")
|
|
|
|
with open(tempname, "rb") as f_loaded:
|
|
with open(f"Tests/fonts/ter-x20b-{encoding}.pil", "rb") as f_target:
|
|
assert f_loaded.read() == f_target.read()
|
|
return tempname
|
|
|
|
|
|
def _test_sanity(request, tmp_path, encoding):
|
|
save_font(request, tmp_path, encoding)
|
|
|
|
|
|
def test_sanity_iso8859_1(request, tmp_path):
|
|
_test_sanity(request, tmp_path, "iso8859-1")
|
|
|
|
|
|
def test_sanity_iso8859_2(request, tmp_path):
|
|
_test_sanity(request, tmp_path, "iso8859-2")
|
|
|
|
|
|
def test_sanity_cp1250(request, tmp_path):
|
|
_test_sanity(request, tmp_path, "cp1250")
|
|
|
|
|
|
def _test_draw(request, tmp_path, encoding):
|
|
tempname = save_font(request, tmp_path, encoding)
|
|
font = ImageFont.load(tempname)
|
|
im = Image.new("L", (150, 30), "white")
|
|
draw = ImageDraw.Draw(im)
|
|
message = charsets[encoding]["message"].encode(encoding)
|
|
draw.text((0, 0), message, "black", font=font)
|
|
assert_image_similar_tofile(im, charsets[encoding]["image1"], 0)
|
|
|
|
|
|
def test_draw_iso8859_1(request, tmp_path):
|
|
_test_draw(request, tmp_path, "iso8859-1")
|
|
|
|
|
|
def test_draw_iso8859_2(request, tmp_path):
|
|
_test_draw(request, tmp_path, "iso8859-2")
|
|
|
|
|
|
def test_draw_cp1250(request, tmp_path):
|
|
_test_draw(request, tmp_path, "cp1250")
|
|
|
|
|
|
def _test_textsize(request, tmp_path, encoding):
|
|
tempname = save_font(request, tmp_path, encoding)
|
|
font = ImageFont.load(tempname)
|
|
for i in range(255):
|
|
(dx, dy) = font.getsize(bytearray([i]))
|
|
assert dy == 20
|
|
assert dx in (0, 10)
|
|
message = charsets[encoding]["message"].encode(encoding)
|
|
for i in range(len(message)):
|
|
msg = message[: i + 1]
|
|
assert font.getsize(msg) == (len(msg) * 10, 20)
|
|
|
|
|
|
def test_textsize_iso8859_1(request, tmp_path):
|
|
_test_textsize(request, tmp_path, "iso8859-1")
|
|
|
|
|
|
def test_textsize_iso8859_2(request, tmp_path):
|
|
_test_textsize(request, tmp_path, "iso8859-2")
|
|
|
|
|
|
def test_textsize_cp1250(request, tmp_path):
|
|
_test_textsize(request, tmp_path, "cp1250")
|