Spaces:
Running on Zero
Running on Zero
File size: 3,915 Bytes
5a57b41 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | """Public-demo input checks and native-resolution rendering behavior."""
from pathlib import Path
import gradio as gr
import numpy as np
from PIL import Image
import pytest
import app
def image_file(tmp_path: Path, size: tuple[int, int], name: str = "image.png") -> str:
path = tmp_path / name
Image.new("RGB", size, color=(60, 80, 100)).save(path)
return str(path)
def test_missing_upload():
with pytest.raises(gr.Error, match="Choose both"):
app.load_image(None)
@pytest.mark.parametrize("size", [(13, 100), (100, 13), (8193, 14)])
def test_image_resource_limits(tmp_path, size):
with pytest.raises(gr.Error):
app.load_image(image_file(tmp_path, size))
def test_upload_pixel_limit_checked_before_decoding(tmp_path, monkeypatch):
monkeypatch.setattr(app, "MAX_UPLOAD_PIXELS", 100)
with pytest.raises(gr.Error, match="24 megapixels"):
app.load_image(image_file(tmp_path, (14, 14)))
def test_gradio_preprocessing_accepts_large_custom_upload(tmp_path):
component = gr.Image(type="filepath")
payload = component.data_model(path=image_file(tmp_path, (1400, 1120)))
path = component.preprocess(payload)
assert app.load_image(path).size == (1400, 1120)
def test_large_aligned_pair_reduced_with_identical_filter():
image = Image.new("RGB", (1400, 1120), color=(60, 80, 100))
reference, distorted = app.prepare_pair(image, image.copy(), True)
assert reference.size == distorted.size
assert reference.width * reference.height <= app.MAX_PIXELS
assert max(reference.size) <= app.MAX_SIDE
assert abs(reference.width / reference.height - 1400 / 1120) < 0.01
assert np.array_equal(np.asarray(reference), np.asarray(distorted))
def test_native_scoring_rejects_large_pair_before_model_load(tmp_path, monkeypatch):
def unexpected_score(*args):
raise AssertionError("Do not score a pair that exceeds the hosted resolution limit")
monkeypatch.setattr(app, "compute_distance_and_map", unexpected_score)
path = image_file(tmp_path, (1400, 1120))
with pytest.raises(gr.Error, match="Enable.*Resize large pairs"):
app.score_pair(path, path, "fast", False)
def test_small_pair_keeps_original_pixels():
image = Image.new("RGB", (31, 43))
reference, distorted = app.prepare_pair(image, image, True)
assert reference is image and distorted is image
def test_resized_result_reports_actual_scoring_grid(tmp_path, monkeypatch):
def score(reference, distorted, variant):
assert reference.size == distorted.size
return 0.25, np.zeros((reference.height // app.PATCH, reference.width // app.PATCH))
monkeypatch.setattr(app, "compute_distance_and_map", score)
path = image_file(tmp_path, (1400, 1120))
value, overlay, details = app.score_pair(path, path, "fast", True)
assert value == 0.25
assert "Demo resize" in details and "1400 × 1120" in details
assert "resized pair" in details
assert overlay.width * overlay.height <= app.MAX_PIXELS
def test_mismatched_native_sizes_rejected_before_metric_load(tmp_path, monkeypatch):
def unexpected_model():
raise AssertionError("Do not load the encoder for an invalid pair")
monkeypatch.setattr(app, "metrics", unexpected_model)
with pytest.raises(gr.Error, match="same size"):
app.score_pair(image_file(tmp_path, (28, 28), "ref.png"),
image_file(tmp_path, (29, 28), "dist.png"), "fast")
def test_overlay_uses_native_center_crop():
reference = Image.new("RGB", (31, 43))
overlay = app.map_overlay(reference, np.zeros((3, 2), dtype=np.float32))
assert overlay.size == (28, 42)
def test_supported_format_conversion_retains_dimensions(tmp_path):
path = tmp_path / "gray.png"
Image.new("L", (31, 43), color=80).save(path)
image = app.load_image(str(path))
assert image.mode == "RGB"
assert image.size == (31, 43)
|