tests: create test to compare multiple image upscalers

imaginairy/enhancers/upscalers/realesrgan.py

@@ -47,7 +47,7 @@ def upscale_image(img, ultrasharp=False, weights_url=None):
     img = img.convert("RGB")
 
     np_img = np.array(img, dtype=np.uint8)
-    upsampler_output, img_mode = realesrgan_upsampler(ultrasharp=ultrasharp, weights_url=weights_url).enhance(
-        np_img[:, :, ::-1]
-    )
+    upsampler_output, img_mode = realesrgan_upsampler(
+        ultrasharp=ultrasharp, weights_url=weights_url
+    ).enhance(np_img[:, :, ::-1])
     return Image.fromarray(upsampler_output[:, :, ::-1], mode=img_mode)

tests/test_enhancers/test_upscalers/test_upscale_realesrgan.py

@@ -15,6 +15,7 @@ def test_upscale_textured_image(filename_base_for_outputs):
         upscaled_image, f"{filename_base_for_outputs}.jpg", threshold=25000
     )
 
+
 @pytest.mark.skip()
 def test_upscalers_difficult_images(filename_base_for_outputs):
     weight_urls = [
@@ -32,7 +33,6 @@ def test_upscalers_difficult_images(filename_base_for_outputs):
         "https://huggingface.co/uwg/upscaler/resolve/main/ESRGAN/4x_NMKDSuperscale_Artisoft_120000_G.pth",
         "https://huggingface.co/uwg/upscaler/resolve/main/ESRGAN/4x_NMKD-Superscale-SP_178000_G.pth",
         "https://huggingface.co/uwg/upscaler/resolve/main/ESRGAN/4xPSNR.pth",
-
     ]
     upscale_challenges_folder = f"{TESTS_FOLDER}/data/upscale_challenges"
     for img_filename in os.listdir(upscale_challenges_folder):
@@ -43,4 +43,6 @@ def test_upscalers_difficult_images(filename_base_for_outputs):
         for url in weight_urls:
             weights_filename = url.split("/")[-1]
             upscaled_image = upscale_image(img, weights_url=url)
-            upscaled_image.save(f"{filename_base_for_outputs}_{img_name}_{weights_filename}.jpg")
+            upscaled_image.save(
+                f"{filename_base_for_outputs}_{img_name}_{weights_filename}.jpg"
+            )