Pillow/Tests/test_font_leaks.py

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)