aboutsummaryrefslogtreecommitdiff
path: root/test/basic_features/txt2img_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/basic_features/txt2img_test.py')
-rw-r--r--test/basic_features/txt2img_test.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/test/basic_features/txt2img_test.py b/test/basic_features/txt2img_test.py
index 1c2674b2..bbc846ed 100644
--- a/test/basic_features/txt2img_test.py
+++ b/test/basic_features/txt2img_test.py
@@ -63,6 +63,10 @@ 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
+ self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200)
+
if __name__ == "__main__":
unittest.main()