pyodide/packages/pyheif/test_pyheif.py

29 lines
776 B
Python

import base64
import pathlib
from pytest_pyodide import run_in_pyodide
DEMO_PATH = pathlib.Path(__file__).parent / "test_data"
SAMPLE_IMAGE = base64.b64encode(
(DEMO_PATH / "tree-with-transparency.heic").read_bytes()
)
def test_read(selenium):
@run_in_pyodide(packages=["pyheif"])
def _test_read_inner(selenium, image_base64):
import base64
with open("tree-with-transparency.heic", "wb") as f:
f.write(base64.b64decode(image_base64))
import pyheif
heif_file = pyheif.read("tree-with-transparency.heic")
assert heif_file.mode == "RGBA"
assert heif_file.size == (262, 264)
assert heif_file.stride == 1056
assert heif_file.bit_depth == 8
_test_read_inner(selenium, SAMPLE_IMAGE)