Spaces:
Paused
Paused
File size: 3,556 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 |
import unittest
import importlib
utils = importlib.import_module('extensions.sd-webui-controlnet.tests.utils', 'utils')
utils.setup_test_env()
import requests
class TestImg2ImgWorkingBase(unittest.TestCase):
def setUp(self):
controlnet_unit = {
"module": "none",
"model": utils.get_model(),
"weight": 1.0,
"input_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,
}
setup_args = {"alwayson_scripts":{"ControlNet":{"args": ([controlnet_unit] * getattr(self, 'units_count', 1))}}}
self.setup_route(setup_args)
def setup_route(self, setup_args):
self.url_img2img = "http://localhost:7860/sdapi/v1/img2img"
self.simple_img2img = {
"init_images": [utils.readImage("test/test_files/img2img_basic.png")],
"resize_mode": 0,
"denoising_strength": 0.75,
"image_cfg_scale": 0,
"mask_blur": 4,
"inpainting_fill": 0,
"inpaint_full_res": True,
"inpaint_full_res_padding": 0,
"inpainting_mask_invert": 0,
"initial_noise_multiplier": 0,
"prompt": "example prompt",
"styles": [],
"seed": -1,
"subseed": -1,
"subseed_strength": 0,
"seed_resize_from_h": -1,
"seed_resize_from_w": -1,
"sampler_name": "Euler a",
"batch_size": 1,
"n_iter": 1,
"steps": 3,
"cfg_scale": 7,
"width": 64,
"height": 64,
"restore_faces": False,
"tiling": False,
"do_not_save_samples": False,
"do_not_save_grid": False,
"negative_prompt": "",
"eta": 0,
"s_churn": 0,
"s_tmax": 0,
"s_tmin": 0,
"s_noise": 1,
"override_settings": {},
"override_settings_restore_afterwards": True,
"sampler_index": "Euler a",
"include_init_images": False,
"send_images": True,
"save_images": False,
"alwayson_scripts": {}
}
self.simple_img2img.update(setup_args)
def assert_status_ok(self):
self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200)
stderr = ""
with open('test/stderr.txt') as f:
stderr = f.read().lower()
with open('test/stderr.txt', 'w') as f:
# clear stderr file so we can easily parse the next test
f.write("")
self.assertFalse('error' in stderr, "Errors in stderr: \n" + stderr)
def test_img2img_simple_performed(self):
self.assert_status_ok()
def test_img2img_alwayson_scripts_default_units(self):
self.units_count = 0
self.setUp()
self.assert_status_ok()
def test_img2img_default_params(self):
self.simple_img2img["alwayson_scripts"]["ControlNet"]["args"] = [{
"input_image": utils.readImage("test/test_files/img2img_basic.png"),
"model": utils.get_model(),
}]
self.assert_status_ok()
if __name__ == "__main__":
unittest.main() |