extras test template added

Author: mezotaken

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()

test/img2img_test.py

@@ -3,7 +3,7 @@
 from gradio.processing_utils import encode_pil_to_base64
 from PIL import Image
 
-class Img2ImgWorking(unittest.TestCase):
+class TestImg2ImgWorking(unittest.TestCase):
     def setUp(self):
         self.url_img2img = "http://localhost:7860/sdapi/v1/img2img"
         self.simple_img2img = {
@@ -56,4 +56,4 @@ class TestImg2ImgCorrectness(unittest.TestCase):
     pass
 
 if __name__ == "__main__":
-    unittest.main()
+    unittest.main()