diff options
author | AUTOMATIC <16777216c@gmail.com> | 2023-01-08 13:14:38 +0000 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-08 13:14:38 +0000 |
commit | 137ce534b2355a527cd1a50c192909161258b442 (patch) | |
tree | c2945d3ca2a08cccb6e9cf5db622ac6f8f41e8a1 /test | |
parent | e7f2f1e1b6f9d616d2b207e596df859d909ab31b (diff) | |
download | stable-diffusion-webui-gfx803-137ce534b2355a527cd1a50c192909161258b442.tar.gz stable-diffusion-webui-gfx803-137ce534b2355a527cd1a50c192909161258b442.tar.bz2 stable-diffusion-webui-gfx803-137ce534b2355a527cd1a50c192909161258b442.zip |
remove some code duplication
remove calls to locals()
add a test for img2img with script
Diffstat (limited to 'test')
-rw-r--r-- | test/basic_features/img2img_test.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/test/basic_features/img2img_test.py b/test/basic_features/img2img_test.py index 0a9c1e8a..bd520b13 100644 --- a/test/basic_features/img2img_test.py +++ b/test/basic_features/img2img_test.py @@ -50,6 +50,12 @@ class TestImg2ImgWorking(unittest.TestCase): self.simple_img2img["mask"] = encode_pil_to_base64(Image.open(r"test/test_files/mask_basic.png")) self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200) + def test_img2img_sd_upscale_performed(self): + self.simple_img2img["script_name"] = "sd upscale" + self.simple_img2img["script_args"] = ["", 8, "Lanczos", 2.0] + + self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200) + if __name__ == "__main__": unittest.main() |