From ffc5b700c417924a77797fc5e02c219f10d801be Mon Sep 17 00:00:00 2001
From: Vladimir Repin <32306715+mezotaken@users.noreply.github.com>
Date: Sat, 29 Oct 2022 21:50:06 +0300
Subject: [PATCH] extras test template added

---
 test/extras_test.py  | 29 +++++++++++++++++++++++++++++
 test/img2img_test.py |  4 ++--
 2 files changed, 31 insertions(+), 2 deletions(-)

diff --git a/test/extras_test.py b/test/extras_test.py
index e69de29b..2e1764d9 100644
--- a/test/extras_test.py
+++ 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()
diff --git a/test/img2img_test.py b/test/img2img_test.py
index d8ed309d..61e3e285 100644
--- a/test/img2img_test.py
+++ b/test/img2img_test.py
@@ -3,7 +3,7 @@ import requests
 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()
\ No newline at end of file
+    unittest.main()