Share formatter cast descriptor policy
This commit is contained in:
@@ -33,6 +33,7 @@ import formatter_input # noqa: E402
|
||||
import hardcore_position_config # noqa: E402
|
||||
import __init__ as sxcp_nodes # noqa: E402
|
||||
import generation_profile_config # noqa: E402
|
||||
import krea_cast # noqa: E402
|
||||
import krea_formatter # noqa: E402
|
||||
import location_config # noqa: E402
|
||||
import prompt_builder as pb # noqa: E402
|
||||
@@ -900,6 +901,35 @@ def smoke_formatter_input_policy() -> None:
|
||||
_expect_text("formatter_input.caption", caption, 20)
|
||||
|
||||
|
||||
def smoke_formatter_cast_policy() -> None:
|
||||
descriptor = (
|
||||
"Woman A / primary creator: 25-year-old adult woman, average figure, warm skin, dark hair; "
|
||||
"Man A: 40-year-old adult man, average figure, tan skin, short dark hair"
|
||||
)
|
||||
entries = [
|
||||
("Woman A", "25-year-old adult woman, average figure, warm skin, dark hair"),
|
||||
("Man A", "40-year-old adult man, average figure, tan skin, short dark hair"),
|
||||
]
|
||||
_expect(krea_cast.cast_entries(descriptor) == entries, "Shared cast entry parser changed")
|
||||
_expect(caption_naturalizer._cast_entries(descriptor) == entries, "Caption cast parser should delegate to shared cast policy")
|
||||
_expect(krea_cast.cast_labels(descriptor) == ["Woman A", "Man A"], "Shared cast label parser changed")
|
||||
_expect(
|
||||
caption_naturalizer._cast_labels(descriptor) == krea_cast.cast_labels(descriptor),
|
||||
"Caption cast labels should delegate to shared cast policy",
|
||||
)
|
||||
natural = krea_cast.natural_cast_descriptor_text(descriptor)
|
||||
_expect(natural.startswith("A 25-year-old adult woman"), "Shared natural cast descriptor text changed")
|
||||
_expect(caption_naturalizer._natural_cast_descriptor_text(descriptor) == natural, "Caption cast descriptor text should delegate")
|
||||
_expect(
|
||||
krea_cast.natural_label_text("Woman A faces Man A.", ["Woman A", "Man A"]) == "The woman faces the man.",
|
||||
"Krea natural label text should keep sentence capitalization",
|
||||
)
|
||||
_expect(
|
||||
caption_naturalizer._natural_label_text("Woman A faces Man A.", ["Woman A", "Man A"]) == "the woman faces the man.",
|
||||
"Caption natural label text should preserve previous lowercase inline behavior",
|
||||
)
|
||||
|
||||
|
||||
def smoke_sdxl_presets_policy() -> None:
|
||||
_expect(
|
||||
sdxl_formatter.SDXL_STYLE_PRESETS is sdxl_presets.SDXL_STYLE_PRESETS,
|
||||
@@ -2912,6 +2942,7 @@ SMOKE_CASES: list[tuple[str, Callable[[], None]]] = [
|
||||
("character_profile_policy", smoke_character_profile_policy),
|
||||
("row_normalization_policy", smoke_row_normalization_policy),
|
||||
("formatter_input_policy", smoke_formatter_input_policy),
|
||||
("formatter_cast_policy", smoke_formatter_cast_policy),
|
||||
("sdxl_presets_policy", smoke_sdxl_presets_policy),
|
||||
("hardcore_position_config_policy", smoke_hardcore_position_config_policy),
|
||||
("category_library_route", smoke_category_library_route),
|
||||
|
||||
Reference in New Issue
Block a user