68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
import unittest
|
|
|
|
try:
|
|
from .test_pro_edit_helpers import FakeMask, FakeModel, FakeTensor, FakeVAE, load_plugin_package, make_conditioning
|
|
except ImportError:
|
|
from test_pro_edit_helpers import FakeMask, FakeModel, FakeTensor, FakeVAE, load_plugin_package, make_conditioning
|
|
|
|
|
|
class TestProEditProfilePrepare(unittest.TestCase):
|
|
def test_prepare_routes_sdxl_base_profile_to_fallback_edit_path(self):
|
|
plugin = load_plugin_package()
|
|
node = plugin.NODE_CLASS_MAPPINGS["LLSProImageEditPrepare"]()
|
|
image = FakeTensor((1, 64, 64, 3), label="image")
|
|
mask = FakeMask((1, 64, 64), mask_bbox=(8, 8, 32, 32), mask_area_ratio=0.2)
|
|
vae = FakeVAE(latent_channels=4, downscale_ratio=8)
|
|
model = FakeModel(
|
|
family="SDXL",
|
|
model_role="base",
|
|
supports_inpaint_native=False,
|
|
supports_image_edit_native=False,
|
|
preferred_edit_backend=None,
|
|
profile_id="sdxl_base",
|
|
backend_type="none",
|
|
sampler_strategy="standard_k",
|
|
loader_strategy="sdxl_checkpoint",
|
|
)
|
|
|
|
latent, _work_image, _work_mask, edit_info, _recommended, positive, negative = node.prepare(
|
|
image=image,
|
|
mask=mask,
|
|
vae=vae,
|
|
positive=make_conditioning("positive"),
|
|
negative=make_conditioning("negative"),
|
|
backend_mode="auto",
|
|
edit_scope="region",
|
|
mask_grow=0,
|
|
mask_blur=0.0,
|
|
mask_threshold=0.5,
|
|
invert_mask=False,
|
|
crop_context=32,
|
|
crop_context_factor=1.0,
|
|
min_size=128,
|
|
max_size=512,
|
|
resize_mode="fit",
|
|
expand_left=0,
|
|
expand_right=0,
|
|
expand_top=0,
|
|
expand_bottom=0,
|
|
canvas_fill="edge",
|
|
auto_recommend="enabled",
|
|
model=model,
|
|
model_info=None,
|
|
)
|
|
|
|
self.assertEqual(edit_info["backend_name"], "sdxl")
|
|
self.assertEqual(edit_info["profile_id"], "sdxl_base")
|
|
self.assertEqual(edit_info["execution_path"], "fallback_repair")
|
|
self.assertIn("noise_mask", latent)
|
|
self.assertEqual(latent["source"], "pro_edit_fallback_region")
|
|
self.assertNotIn("concat_latent_image", positive[0][1])
|
|
self.assertNotIn("concat_mask", positive[0][1])
|
|
self.assertNotIn("concat_latent_image", negative[0][1])
|
|
self.assertNotIn("concat_mask", negative[0][1])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|