mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-26 09:56:17 +03:00
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from PIL import Image, ImageDraw, ImageFont, _util
|
|
|
|
from .helper import PillowLeakTestCase, features, skip_unless_feature
|
|
|
|
original_core = ImageFont.core
|
|
|
|
|
|
class TestTTypeFontLeak(PillowLeakTestCase):
|
|
# fails at iteration 3 in main
|
|
iterations = 10
|
|
mem_limit = 4096 # k
|
|
|
|
def _test_font(self, font: ImageFont.FreeTypeFont) -> None:
|
|
im = Image.new("RGB", (255, 255), "white")
|
|
draw = ImageDraw.ImageDraw(im)
|
|
self._test_leak(
|
|
lambda: draw.text(
|
|
(0, 0), "some text " * 1024, font=font, fill="black" # ~10k
|
|
)
|
|
)
|
|
|
|
@skip_unless_feature("freetype2")
|
|
def test_leak(self) -> None:
|
|
ttype = ImageFont.truetype("Tests/fonts/FreeMono.ttf", 20)
|
|
self._test_font(ttype)
|
|
|
|
|
|
class TestDefaultFontLeak(TestTTypeFontLeak):
|
|
# fails at iteration 37 in main
|
|
iterations = 100
|
|
mem_limit = 1024 # k
|
|
|
|
def test_leak(self) -> None:
|
|
if features.check_module("freetype2"):
|
|
ImageFont.core = _util.DeferredError(ImportError)
|
|
try:
|
|
default_font = ImageFont.load_default()
|
|
finally:
|
|
ImageFont.core = original_core
|
|
|
|
self._test_font(default_font)
|