Share fallback field-label cleanup
This commit is contained in:
@@ -904,9 +904,19 @@ def smoke_formatter_input_policy() -> None:
|
||||
formatter_input.prompt_field(labeled_prompt, "Camera control") == "side view",
|
||||
"Shared formatter field-label inventory lost Camera control parsing",
|
||||
)
|
||||
stripped_labels = formatter_input.strip_prompt_field_labels(
|
||||
"Characters: woman. Erotic outfit: sheer dress. Camera: side view."
|
||||
)
|
||||
_expect("Characters:" not in stripped_labels, "Shared label stripper did not remove Characters label")
|
||||
_expect("Erotic outfit:" not in stripped_labels, "Shared label stripper did not remove Erotic outfit label")
|
||||
_expect("Camera:" not in stripped_labels, "Shared label stripper did not remove Camera label")
|
||||
|
||||
_expect(krea_formatter._clean("a b , c") == formatter_input.clean_text("a b , c"), "Krea clean helper is not delegated")
|
||||
_expect(sdxl_formatter._clean("a b , c") == formatter_input.clean_text("a b , c"), "SDXL clean helper is not delegated")
|
||||
_expect(
|
||||
sdxl_formatter._strip_prompt_field_labels("Characters: woman. Camera: side view.") == "woman. side view.",
|
||||
"SDXL label stripper should delegate to formatter_input",
|
||||
)
|
||||
_expect(caption_naturalizer._clean_text("a b , c") == formatter_input.clean_text("a b , c"), "Caption clean helper is not delegated")
|
||||
_expect(krea_formatter._strip_trigger(f"{Trigger}, prompt text", False) == "prompt text", "Krea trigger stripping changed")
|
||||
_expect(sdxl_formatter._strip_trigger(f"{SdxlTrigger}, prompt text", False) == "prompt text", "SDXL trigger stripping changed")
|
||||
@@ -921,6 +931,19 @@ def smoke_formatter_input_policy() -> None:
|
||||
_expect_text("formatter_input.krea_prompt", krea.get("krea_prompt"), 20)
|
||||
_expect_text("formatter_input.sdxl_prompt", sdxl.get("sdxl_prompt"), 20)
|
||||
_expect_text("formatter_input.caption", caption, 20)
|
||||
fallback_sdxl = sdxl_formatter.format_sdxl_prompt(
|
||||
"Characters: woman. Erotic outfit: sheer dress. Camera: side view. Avoid: blur",
|
||||
input_hint="prompt",
|
||||
style_preset="none",
|
||||
quality_preset="none",
|
||||
trigger=SdxlTrigger,
|
||||
prepend_trigger=False,
|
||||
)
|
||||
fallback_prompt = fallback_sdxl.get("sdxl_prompt", "")
|
||||
_expect("Characters:" not in fallback_prompt, "SDXL fallback leaked Characters label")
|
||||
_expect("Erotic outfit:" not in fallback_prompt, "SDXL fallback leaked Erotic outfit label")
|
||||
_expect("Camera:" not in fallback_prompt, "SDXL fallback leaked Camera label")
|
||||
_expect("blur" in fallback_sdxl.get("negative_prompt", ""), "SDXL fallback lost Avoid negative text")
|
||||
|
||||
|
||||
def smoke_formatter_cast_policy() -> None:
|
||||
|
||||
Reference in New Issue
Block a user