1 files changed, 29 insertions, 0 deletions
diff --git a/test/extras_test.py b/test/extras_test.py
new file mode 100644
index 00000000..2e1764d9
--- /dev/null
+++ b/test/extras_test.py
@@ -0,0 +1,29 @@
+import unittest
+import requests
+from gradio.processing_utils import encode_pil_to_base64
+from PIL import Image
+
+class TestExtrasWorking(unittest.TestCase):
+    def setUp(self):
+        self.url_img2img = "http://localhost:7860/sdapi/v1/extra-single-image"
+        self.simple_extras = {
+            "resize_mode": 0,
+            "show_extras_results": True,
+            "gfpgan_visibility": 0,
+            "codeformer_visibility": 0,
+            "codeformer_weight": 0,
+            "upscaling_resize": 2,
+            "upscaling_resize_w": 512,
+            "upscaling_resize_h": 512,
+            "upscaling_crop": True,
+            "upscaler_1": "None",
+            "upscaler_2": "None",
+            "extras_upscaler_2_visibility": 0,
+            "image": ""
+            }
+
+class TestExtrasCorrectness(unittest.TestCase):
+    pass
+
+if __name__ == "__main__":
+    unittest.main()