| import unittest | |
| import requests | |
| class TestTxt2ImgWorking(unittest.TestCase): | |
| def setUp(self): | |
| self.url_txt2img = "http://localhost:7860/sdapi/v1/txt2img" | |
| self.simple_txt2img = { | |
| "enable_hr": False, | |
| "denoising_strength": 0, | |
| "firstphase_width": 0, | |
| "firstphase_height": 0, | |
| "prompt": "example prompt", | |
| "styles": [], | |
| "seed": -1, | |
| "subseed": -1, | |
| "subseed_strength": 0, | |
| "seed_resize_from_h": -1, | |
| "seed_resize_from_w": -1, | |
| "batch_size": 1, | |
| "n_iter": 1, | |
| "steps": 3, | |
| "cfg_scale": 7, | |
| "width": 64, | |
| "height": 64, | |
| "restore_faces": False, | |
| "tiling": False, | |
| "negative_prompt": "", | |
| "eta": 0, | |
| "s_churn": 0, | |
| "s_tmax": 0, | |
| "s_tmin": 0, | |
| "s_noise": 1, | |
| "sampler_index": "Euler a" | |
| } | |
| def test_txt2img_simple_performed(self): | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| def test_txt2img_with_negative_prompt_performed(self): | |
| self.simple_txt2img["negative_prompt"] = "example negative prompt" | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| def test_txt2img_with_complex_prompt_performed(self): | |
| self.simple_txt2img["prompt"] = "((emphasis)), (emphasis1:1.1), [to:1], [from::2], [from:to:0.3], [alt|alt1]" | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| def test_txt2img_not_square_image_performed(self): | |
| self.simple_txt2img["height"] = 128 | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| def test_txt2img_with_hrfix_performed(self): | |
| self.simple_txt2img["enable_hr"] = True | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| def test_txt2img_with_tiling_performed(self): | |
| self.simple_txt2img["tiling"] = True | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| def test_txt2img_with_restore_faces_performed(self): | |
| self.simple_txt2img["restore_faces"] = True | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| def test_txt2img_with_vanilla_sampler_performed(self): | |
| self.simple_txt2img["sampler_index"] = "PLMS" | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| self.simple_txt2img["sampler_index"] = "DDIM" | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| self.simple_txt2img["sampler_index"] = "UniPC" | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| def test_txt2img_multiple_batches_performed(self): | |
| self.simple_txt2img["n_iter"] = 2 | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| def test_txt2img_batch_performed(self): | |
| self.simple_txt2img["batch_size"] = 2 | |
| self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) | |
| if __name__ == "__main__": | |
| unittest.main() | |