diff options
| author | Vladimir Repin <32306715+mezotaken@users.noreply.github.com> | 2023-01-09 18:01:28 +0000 |
|---|---|---|
| committer | Vladimir Repin <32306715+mezotaken@users.noreply.github.com> | 2023-01-09 18:01:28 +0000 |
| commit | 00005ac9af10d58a75f7ce0aa04db78775808e93 (patch) | |
| tree | 63daeba6b2269fd33196f73178b2f2255266da9a /test/basic_features/txt2img_test.py | |
| parent | 7d2bb86cce10ee6a8e81aaad810544a4ca38cec9 (diff) | |
| download | stable-diffusion-webui-gfx803-00005ac9af10d58a75f7ce0aa04db78775808e93.tar.gz stable-diffusion-webui-gfx803-00005ac9af10d58a75f7ce0aa04db78775808e93.tar.bz2 stable-diffusion-webui-gfx803-00005ac9af10d58a75f7ce0aa04db78775808e93.zip | |
add more tests
Diffstat (limited to 'test/basic_features/txt2img_test.py')
| -rw-r--r-- | test/basic_features/txt2img_test.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/test/basic_features/txt2img_test.py b/test/basic_features/txt2img_test.py index bbc846ed..5b27a7ec 100644 --- a/test/basic_features/txt2img_test.py +++ b/test/basic_features/txt2img_test.py @@ -41,6 +41,9 @@ class TestTxt2ImgWorking(unittest.TestCase): 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]" + 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) @@ -53,6 +56,10 @@ class TestTxt2ImgWorking(unittest.TestCase): 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) @@ -63,8 +70,8 @@ class TestTxt2ImgWorking(unittest.TestCase): self.simple_txt2img["n_iter"] = 2 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 + 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) |