Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-14 00:58:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add test_multi_upscaler_small to check that MultiUpscaler works with small (16x16px) images

Browse source
This commit is contained in:
Laurent 2024-10-01 08:56:40 +00:00 committed by Laureηt
parent 7b6f97347e
commit f89c4f720d
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/e2e/test_diffusion.py
View file

@ -2675,6 +2675,16 @@ def test_multi_upscaler(
ensure_similar_images(predicted_image, expected_multi_upscaler, min_psnr=35, min_ssim=0.99) ensure_similar_images(predicted_image, expected_multi_upscaler, min_psnr=35, min_ssim=0.99)
@no_grad()
def test_multi_upscaler_small(
multi_upscaler: MultiUpscaler,
clarity_example: Image.Image,
) -> None:
image = clarity_example.resize((16, 16))
image = multi_upscaler.upscale(image) # check we can upscale a small image
image = multi_upscaler.upscale(image) # check we can upscale it twice
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
def expected_ic_light(ref_path: Path) -> Image.Image: def expected_ic_light(ref_path: Path) -> Image.Image:
return _img_open(ref_path / "expected_ic_light.png").convert("RGB") return _img_open(ref_path / "expected_ic_light.png").convert("RGB")