diff options
| author | Vladimir Repin <32306715+mezotaken@users.noreply.github.com> | 2022-11-12 15:12:15 +0300 |
|---|---|---|
| committer | Vladimir Repin <32306715+mezotaken@users.noreply.github.com> | 2022-11-12 15:12:15 +0300 |
| commit | 007f4f7314eabd9cc3a2b0d11889de49ad3c682a (patch) | |
| tree | d62f8156fcb992a069892e75865d318d03c8cb29 /test/txt2img_test.py | |
| parent | 98947d173e3f1667eba29c904f681047dea9de90 (diff) | |
Tests cleaned up
Diffstat (limited to 'test/txt2img_test.py')
| -rw-r--r-- | test/txt2img_test.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/test/txt2img_test.py b/test/txt2img_test.py index 1936e07e..ce752085 100644 --- a/test/txt2img_test.py +++ b/test/txt2img_test.py @@ -53,13 +53,15 @@ class TestTxt2ImgWorking(unittest.TestCase): self.simple_txt2img["restore_faces"] = True self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) - def test_txt2img_with_tiling_faces_performed(self): + 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_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) def test_txt2img_multiple_batches_performed(self): self.simple_txt2img["n_iter"] = 2 |