Pillow

1
from __future__ import annotations
2

3
from io import BytesIO
4

5
from PIL import Image
6

7
from .helper import PillowLeakTestCase, skip_unless_feature
8

9
test_file = "Tests/images/hopper.webp"
10

11

12
@skip_unless_feature("webp")
13
class TestWebPLeaks(PillowLeakTestCase):
14
    mem_limit = 3 * 1024  # kb
15
    iterations = 100
16

17
    def test_leak_load(self) -> None:
18
        with open(test_file, "rb") as f:
19
            im_data = f.read()
20

21
        def core() -> None:
22
            with Image.open(BytesIO(im_data)) as im:
23
                im.load()
24

25
        self._test_leak(core)
26