StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Reindent utils_test with 4 spaces

Browse Source
This commit is contained in:
Aarni Koskela 2023-05-11 18:13:54 +03:00
parent 098d2fda52
commit 431bc5a297
1 changed files with 44 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
test/basic_features/utils_test.py
View File

@ -1,62 +1,64 @@
import unittest import unittest
import requests import requests
class UtilsTests(unittest.TestCase): class UtilsTests(unittest.TestCase):
def setUp(self): def setUp(self):
self.url_options = "http://localhost:7860/sdapi/v1/options" self.url_options = "http://localhost:7860/sdapi/v1/options"
self.url_cmd_flags = "http://localhost:7860/sdapi/v1/cmd-flags" self.url_cmd_flags = "http://localhost:7860/sdapi/v1/cmd-flags"
self.url_samplers = "http://localhost:7860/sdapi/v1/samplers" self.url_samplers = "http://localhost:7860/sdapi/v1/samplers"
self.url_upscalers = "http://localhost:7860/sdapi/v1/upscalers" self.url_upscalers = "http://localhost:7860/sdapi/v1/upscalers"
self.url_sd_models = "http://localhost:7860/sdapi/v1/sd-models" self.url_sd_models = "http://localhost:7860/sdapi/v1/sd-models"
self.url_hypernetworks = "http://localhost:7860/sdapi/v1/hypernetworks" self.url_hypernetworks = "http://localhost:7860/sdapi/v1/hypernetworks"
self.url_face_restorers = "http://localhost:7860/sdapi/v1/face-restorers" self.url_face_restorers = "http://localhost:7860/sdapi/v1/face-restorers"
self.url_realesrgan_models = "http://localhost:7860/sdapi/v1/realesrgan-models" self.url_realesrgan_models = "http://localhost:7860/sdapi/v1/realesrgan-models"
self.url_prompt_styles = "http://localhost:7860/sdapi/v1/prompt-styles" self.url_prompt_styles = "http://localhost:7860/sdapi/v1/prompt-styles"
self.url_embeddings = "http://localhost:7860/sdapi/v1/embeddings" self.url_embeddings = "http://localhost:7860/sdapi/v1/embeddings"
def test_options_get(self): def test_options_get(self):
self.assertEqual(requests.get(self.url_options).status_code, 200) self.assertEqual(requests.get(self.url_options).status_code, 200)
def test_options_write(self): def test_options_write(self):
response = requests.get(self.url_options) response = requests.get(self.url_options)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
pre_value = response.json()["send_seed"] pre_value = response.json()["send_seed"]
self.assertEqual(requests.post(self.url_options, json={"send_seed":not pre_value}).status_code, 200) self.assertEqual(requests.post(self.url_options, json={"send_seed": not pre_value}).status_code, 200)
response = requests.get(self.url_options) response = requests.get(self.url_options)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(response.json()["send_seed"], not pre_value) self.assertEqual(response.json()["send_seed"], not pre_value)
requests.post(self.url_options, json={"send_seed": pre_value}) requests.post(self.url_options, json={"send_seed": pre_value})
def test_cmd_flags(self): def test_cmd_flags(self):
self.assertEqual(requests.get(self.url_cmd_flags).status_code, 200) self.assertEqual(requests.get(self.url_cmd_flags).status_code, 200)
def test_samplers(self): def test_samplers(self):
self.assertEqual(requests.get(self.url_samplers).status_code, 200) self.assertEqual(requests.get(self.url_samplers).status_code, 200)
def test_upscalers(self): def test_upscalers(self):
self.assertEqual(requests.get(self.url_upscalers).status_code, 200) self.assertEqual(requests.get(self.url_upscalers).status_code, 200)
def test_sd_models(self): def test_sd_models(self):
self.assertEqual(requests.get(self.url_sd_models).status_code, 200) self.assertEqual(requests.get(self.url_sd_models).status_code, 200)
def test_hypernetworks(self): def test_hypernetworks(self):
self.assertEqual(requests.get(self.url_hypernetworks).status_code, 200) self.assertEqual(requests.get(self.url_hypernetworks).status_code, 200)
def test_face_restorers(self): def test_face_restorers(self):
self.assertEqual(requests.get(self.url_face_restorers).status_code, 200) self.assertEqual(requests.get(self.url_face_restorers).status_code, 200)
def test_realesrgan_models(self): def test_realesrgan_models(self):
self.assertEqual(requests.get(self.url_realesrgan_models).status_code, 200) self.assertEqual(requests.get(self.url_realesrgan_models).status_code, 200)
def test_prompt_styles(self): def test_prompt_styles(self):
self.assertEqual(requests.get(self.url_prompt_styles).status_code, 200) 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__": if __name__ == "__main__":
unittest.main() unittest.main()