mirror of
https://github.com/python-pillow/Pillow.git
synced 2026-01-28 02:55:49 +03:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from PIL import Image, ImageDraw, ImageFont, _util
|
|
|
|
from .helper import PillowLeakTestCase, features, skip_unless_feature
|
|
|
|
original_core = ImageFont.core
|
|
|
|
|
|
class TestFontLeak(PillowLeakTestCase):
|
|
def _test_font(self, font: ImageFont.FreeTypeFont | ImageFont.ImageFont) -> 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
|
|
)
|
|
)
|
|
|
|
|
|
class TestTTypeFontLeak(TestFontLeak):
|
|
# fails at iteration 3 in main
|
|
iterations = 10
|
|
mem_limit = 4096 # k
|
|
|
|
@skip_unless_feature("freetype2")
|
|
def test_leak(self) -> None:
|
|
ttype = ImageFont.truetype("Tests/fonts/FreeMono.ttf", 20)
|
|
self._test_font(ttype)
|
|
|
|
|
|
class TestDefaultFontLeak(TestFontLeak):
|
|
# fails at iteration 37 in main
|
|
iterations = 100
|
|
mem_limit = 1024 # k
|
|
|
|
def test_leak(self, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
if features.check_module("freetype2"):
|
|
monkeypatch.setattr(
|
|
ImageFont,
|
|
"core",
|
|
_util.DeferredError(ImportError("Disabled for testing")),
|
|
)
|
|
default_font = ImageFont.load_default()
|
|
self._test_font(default_font)
|