Spaces:
Paused
Paused
File size: 5,148 Bytes
3f9c56c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 |
import unittest
import importlib
utils = importlib.import_module('extensions.sd-webui-controlnet.tests.utils', 'utils')
utils.setup_test_env()
import requests
class TestAlwaysonTxt2ImgWorking(unittest.TestCase):
def setUp(self):
controlnet_unit = {
"enabled": True,
"module": "none",
"model": utils.get_model(),
"weight": 1.0,
"image": utils.readImage("test/test_files/img2img_basic.png"),
"mask": utils.readImage("test/test_files/img2img_basic.png"),
"resize_mode": 1,
"lowvram": False,
"processor_res": 64,
"threshold_a": 64,
"threshold_b": 64,
"guidance_start": 0.0,
"guidance_end": 1.0,
"control_mode": 0,
"pixel_perfect": False
}
setup_args = [controlnet_unit] * getattr(self, 'units_count', 1)
self.setup_route(setup_args)
def setup_route(self, setup_args):
self.url_txt2img = "http://localhost:7860/sdapi/v1/txt2img"
self.simple_txt2img = {
"enable_hr": False,
"denoising_strength": 0,
"firstphase_width": 0,
"firstphase_height": 0,
"prompt": "example prompt",
"styles": [],
"seed": -1,
"subseed": -1,
"subseed_strength": 0,
"seed_resize_from_h": -1,
"seed_resize_from_w": -1,
"batch_size": 1,
"n_iter": 1,
"steps": 3,
"cfg_scale": 7,
"width": 64,
"height": 64,
"restore_faces": False,
"tiling": False,
"negative_prompt": "",
"eta": 0,
"s_churn": 0,
"s_tmax": 0,
"s_tmin": 0,
"s_noise": 1,
"sampler_index": "Euler a",
"alwayson_scripts": {}
}
self.setup_controlnet_params(setup_args)
def setup_controlnet_params(self, setup_args):
self.simple_txt2img["alwayson_scripts"]["ControlNet"] = {
"args": setup_args
}
def assert_status_ok(self, msg=None):
self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200, msg)
stderr = ""
with open('test/stderr.txt') as f:
stderr = f.read().lower()
with open('test/stderr.txt', 'w') as f:
# clear stderr file so that we can easily parse the next test
f.write("")
self.assertFalse('error' in stderr, "Errors in stderr: \n" + stderr)
def test_txt2img_simple_performed(self):
self.assert_status_ok()
def test_txt2img_alwayson_scripts_default_units(self):
self.units_count = 0
self.setUp()
self.assert_status_ok()
def test_txt2img_multiple_batches_performed(self):
self.simple_txt2img["n_iter"] = 2
self.assert_status_ok()
def test_txt2img_batch_performed(self):
self.simple_txt2img["batch_size"] = 2
self.assert_status_ok()
def test_txt2img_2_units(self):
self.units_count = 2
self.setUp()
self.assert_status_ok()
def test_txt2img_8_units(self):
self.units_count = 8
self.setUp()
self.assert_status_ok()
def test_txt2img_default_params(self):
self.simple_txt2img["alwayson_scripts"]["ControlNet"]["args"] = [
{
"input_image": utils.readImage("test/test_files/img2img_basic.png"),
"model": utils.get_model(),
}
]
self.assert_status_ok()
def test_call_with_preprocessors(self):
available_modules = utils.get_modules()
available_modules_list = available_modules.get('module_list', [])
available_modules_detail = available_modules.get('module_detail', {})
for module in ['depth', 'openpose_full']:
assert module in available_modules_list, f'Failed to find {module}.'
assert module in available_modules_detail, f"Failed to find {module}'s detail."
with self.subTest(module=module):
self.simple_txt2img["alwayson_scripts"]["ControlNet"]["args"] = [
{
"input_image": utils.readImage("test/test_files/img2img_basic.png"),
"model": utils.get_model(),
"module": module
}
]
self.assert_status_ok(f'Running preprocessor module: {module}')
def test_call_invalid_params(self):
for param in ('processor_res', 'threshold_a', 'threshold_b'):
with self.subTest(param=param):
self.simple_txt2img["alwayson_scripts"]["ControlNet"]["args"] = [
{
"input_image": utils.readImage("test/test_files/img2img_basic.png"),
"model": utils.get_model(),
param: -1,
}
]
self.assert_status_ok(f'Run with {param} = -1.')
if __name__ == "__main__":
unittest.main()
|