wehub-resource-sync
32 lines
975 B
Python
32 lines
975 B
Python
from __future__ import annotations
|
|
|
|
import base64
|
|
import io
|
|
|
|
from PIL import Image
|
|
|
|
from yuxi.utils.image_processor import process_uploaded_image
|
|
|
|
|
|
def test_process_uploaded_image_composites_transparent_png_pixels_on_white():
|
|
image = Image.new("RGBA", (2, 2), (255, 255, 255, 0))
|
|
image.putpixel((0, 0), (50, 87, 244, 0))
|
|
image.putpixel((1, 0), (50, 87, 244, 255))
|
|
|
|
with io.BytesIO() as buffer:
|
|
image.save(buffer, format="PNG")
|
|
image_data = buffer.getvalue()
|
|
|
|
result = process_uploaded_image(image_data, "transparent.png")
|
|
|
|
assert result["success"] is True
|
|
assert result["format"] == "PNG"
|
|
assert result["mime_type"] == "image/png"
|
|
|
|
processed_data = base64.b64decode(result["image_content"])
|
|
with Image.open(io.BytesIO(processed_data)) as processed_image:
|
|
rgb_image = processed_image.convert("RGB")
|
|
|
|
assert rgb_image.getpixel((0, 0)) == (255, 255, 255)
|
|
assert rgb_image.getpixel((1, 0)) == (50, 87, 244)
|