#!/usr/bin/python3
"""Decode to YUV buffer and report plane sizes."""

import numpy as np
from turbojpeg import TurboJPEG

jpeg = TurboJPEG()
img = np.full((16, 16, 3), 200, dtype=np.uint8)
jpeg_bytes = jpeg.encode(img)

buffer, plane_sizes = jpeg.decode_to_yuv(jpeg_bytes)

assert isinstance(plane_sizes, list)
print("plane_sizes:", plane_sizes)

