Merge branch 'dev' of https://github.com/AUTOMATIC1111/stable-diffusion-webui into tomesd

1 files changed, 44 insertions, 42 deletions

diff --git a/test/basic_features/utils_test.py b/test/basic_features/utils_test.py
index 0bfc28a0..d9e46b5e 100644
--- a/test/basic_features/utils_test.py
+++ b/test/basic_features/utils_test.py
@@ -1,62 +1,64 @@
 import unittest
 import requests
 
+
 class UtilsTests(unittest.TestCase):
-  def setUp(self):
-    self.url_options = "http://localhost:7860/sdapi/v1/options"
-    self.url_cmd_flags = "http://localhost:7860/sdapi/v1/cmd-flags"
-    self.url_samplers = "http://localhost:7860/sdapi/v1/samplers"
-    self.url_upscalers = "http://localhost:7860/sdapi/v1/upscalers"
-    self.url_sd_models = "http://localhost:7860/sdapi/v1/sd-models"
-    self.url_hypernetworks = "http://localhost:7860/sdapi/v1/hypernetworks"
-    self.url_face_restorers = "http://localhost:7860/sdapi/v1/face-restorers"
-    self.url_realesrgan_models = "http://localhost:7860/sdapi/v1/realesrgan-models"
-    self.url_prompt_styles = "http://localhost:7860/sdapi/v1/prompt-styles"
-    self.url_embeddings = "http://localhost:7860/sdapi/v1/embeddings"
+    def setUp(self):
+        self.url_options = "http://localhost:7860/sdapi/v1/options"
+        self.url_cmd_flags = "http://localhost:7860/sdapi/v1/cmd-flags"
+        self.url_samplers = "http://localhost:7860/sdapi/v1/samplers"
+        self.url_upscalers = "http://localhost:7860/sdapi/v1/upscalers"
+        self.url_sd_models = "http://localhost:7860/sdapi/v1/sd-models"
+        self.url_hypernetworks = "http://localhost:7860/sdapi/v1/hypernetworks"
+        self.url_face_restorers = "http://localhost:7860/sdapi/v1/face-restorers"
+        self.url_realesrgan_models = "http://localhost:7860/sdapi/v1/realesrgan-models"
+        self.url_prompt_styles = "http://localhost:7860/sdapi/v1/prompt-styles"
+        self.url_embeddings = "http://localhost:7860/sdapi/v1/embeddings"
+
+    def test_options_get(self):
+        self.assertEqual(requests.get(self.url_options).status_code, 200)
+
+    def test_options_write(self):
+        response = requests.get(self.url_options)
+        self.assertEqual(response.status_code, 200)
+
+        pre_value = response.json()["send_seed"]
 
-  def test_options_get(self):
-    self.assertEqual(requests.get(self.url_options).status_code, 200)
+        self.assertEqual(requests.post(self.url_options, json={"send_seed": not pre_value}).status_code, 200)
 
-  def test_options_write(self):
-    response = requests.get(self.url_options)
-    self.assertEqual(response.status_code, 200)
+        response = requests.get(self.url_options)
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.json()["send_seed"], not pre_value)
 
-    pre_value = response.json()["send_seed"]
+        requests.post(self.url_options, json={"send_seed": pre_value})
 
-    self.assertEqual(requests.post(self.url_options, json={"send_seed":not pre_value}).status_code, 200)
+    def test_cmd_flags(self):
+        self.assertEqual(requests.get(self.url_cmd_flags).status_code, 200)
 
-    response = requests.get(self.url_options)
-    self.assertEqual(response.status_code, 200)
-    self.assertEqual(response.json()["send_seed"], not pre_value)
+    def test_samplers(self):
+        self.assertEqual(requests.get(self.url_samplers).status_code, 200)
 
-    requests.post(self.url_options, json={"send_seed": pre_value})
+    def test_upscalers(self):
+        self.assertEqual(requests.get(self.url_upscalers).status_code, 200)
 
-  def test_cmd_flags(self):
-    self.assertEqual(requests.get(self.url_cmd_flags).status_code, 200)
+    def test_sd_models(self):
+        self.assertEqual(requests.get(self.url_sd_models).status_code, 200)
 
-  def test_samplers(self):
-    self.assertEqual(requests.get(self.url_samplers).status_code, 200)
+    def test_hypernetworks(self):
+        self.assertEqual(requests.get(self.url_hypernetworks).status_code, 200)
 
-  def test_upscalers(self):
-    self.assertEqual(requests.get(self.url_upscalers).status_code, 200)
+    def test_face_restorers(self):
+        self.assertEqual(requests.get(self.url_face_restorers).status_code, 200)
 
-  def test_sd_models(self):
-    self.assertEqual(requests.get(self.url_sd_models).status_code, 200)
+    def test_realesrgan_models(self):
+        self.assertEqual(requests.get(self.url_realesrgan_models).status_code, 200)
 
-  def test_hypernetworks(self):
-    self.assertEqual(requests.get(self.url_hypernetworks).status_code, 200)
+    def test_prompt_styles(self):
+        self.assertEqual(requests.get(self.url_prompt_styles).status_code, 200)
 
-  def test_face_restorers(self):
-    self.assertEqual(requests.get(self.url_face_restorers).status_code, 200)
-  
-  def test_realesrgan_models(self):
-    self.assertEqual(requests.get(self.url_realesrgan_models).status_code, 200)
-  
-  def test_prompt_styles(self):
-    self.assertEqual(requests.get(self.url_prompt_styles).status_code, 200)
+    def test_embeddings(self):
+        self.assertEqual(requests.get(self.url_embeddings).status_code, 200)
 
-  def test_embeddings(self):
-    self.assertEqual(requests.get(self.url_embeddings).status_code, 200)
 
 if __name__ == "__main__":
     unittest.main()