#!/usr/bin/python3
"""Test scale_with_quality() from README examples."""

import numpy as np
from turbojpeg import TurboJPEG

jpeg = TurboJPEG()
img = np.ones((32, 32, 3), dtype=np.uint8) * 127
jpeg_bytes = jpeg.encode(img, quality=90)

scaled_bytes = jpeg.scale_with_quality(jpeg_bytes, scaling_factor=(1, 2), quality=70)
scaled = jpeg.decode(scaled_bytes)

assert scaled.shape[0] <= 32 and scaled.shape[1] <= 32
print("scaled shape:", scaled.shape)

