diff options
author | Aarni Koskela <akx@iki.fi> | 2023-05-11 18:57:43 +0000 |
---|---|---|
committer | Aarni Koskela <akx@iki.fi> | 2023-05-19 14:42:34 +0000 |
commit | 793a491923ba11adbe1024e0eb0402923165dafa (patch) | |
tree | bf070c28ba3fff59e83e4bcbd3a0adb6e57f112f /test/basic_features/extras_test.py | |
parent | d41a31a50805fa70f7299f5c0384ad41db7142e5 (diff) | |
download | stable-diffusion-webui-gfx803-793a491923ba11adbe1024e0eb0402923165dafa.tar.gz stable-diffusion-webui-gfx803-793a491923ba11adbe1024e0eb0402923165dafa.tar.bz2 stable-diffusion-webui-gfx803-793a491923ba11adbe1024e0eb0402923165dafa.zip |
Overhaul tests to use py.test
Diffstat (limited to 'test/basic_features/extras_test.py')
-rw-r--r-- | test/basic_features/extras_test.py | 56 |
1 files changed, 0 insertions, 56 deletions
diff --git a/test/basic_features/extras_test.py b/test/basic_features/extras_test.py deleted file mode 100644 index 8ed98747..00000000 --- a/test/basic_features/extras_test.py +++ /dev/null @@ -1,56 +0,0 @@ -import os -import unittest -import requests -from gradio.processing_utils import encode_pil_to_base64 -from PIL import Image -from modules.paths import script_path - -class TestExtrasWorking(unittest.TestCase): - def setUp(self): - self.url_extras_single = "http://localhost:7860/sdapi/v1/extra-single-image" - self.extras_single = { - "resize_mode": 0, - "show_extras_results": True, - "gfpgan_visibility": 0, - "codeformer_visibility": 0, - "codeformer_weight": 0, - "upscaling_resize": 2, - "upscaling_resize_w": 128, - "upscaling_resize_h": 128, - "upscaling_crop": True, - "upscaler_1": "None", - "upscaler_2": "None", - "extras_upscaler_2_visibility": 0, - "image": encode_pil_to_base64(Image.open(os.path.join(script_path, r"test/test_files/img2img_basic.png"))) - } - - def test_simple_upscaling_performed(self): - self.extras_single["upscaler_1"] = "Lanczos" - self.assertEqual(requests.post(self.url_extras_single, json=self.extras_single).status_code, 200) - - -class TestPngInfoWorking(unittest.TestCase): - def setUp(self): - self.url_png_info = "http://localhost:7860/sdapi/v1/extra-single-image" - self.png_info = { - "image": encode_pil_to_base64(Image.open(os.path.join(script_path, r"test/test_files/img2img_basic.png"))) - } - - def test_png_info_performed(self): - self.assertEqual(requests.post(self.url_png_info, json=self.png_info).status_code, 200) - - -class TestInterrogateWorking(unittest.TestCase): - def setUp(self): - self.url_interrogate = "http://localhost:7860/sdapi/v1/extra-single-image" - self.interrogate = { - "image": encode_pil_to_base64(Image.open(os.path.join(script_path, r"test/test_files/img2img_basic.png"))), - "model": "clip" - } - - def test_interrogate_performed(self): - self.assertEqual(requests.post(self.url_interrogate, json=self.interrogate).status_code, 200) - - -if __name__ == "__main__": - unittest.main() |