Move pair cast styling policy
This commit is contained in:
@@ -41,6 +41,7 @@ import krea_cast # noqa: E402
|
||||
import krea_formatter # noqa: E402
|
||||
import location_config # noqa: E402
|
||||
import loop_nodes # noqa: E402
|
||||
import pair_cast # noqa: E402
|
||||
import pair_clothing # noqa: E402
|
||||
import prompt_builder as pb # noqa: E402
|
||||
import pov_policy # noqa: E402
|
||||
@@ -1681,6 +1682,27 @@ def smoke_pair_options_policy() -> None:
|
||||
== ["Woman A's body is fully exposed, bare skin unobstructed"],
|
||||
"Pair clothing character entries should skip POV labels",
|
||||
)
|
||||
_expect(
|
||||
pair_cast.cast_summary_phrase(2, 1) == "2 women, 1 man, 3 total adults",
|
||||
"Pair cast summary phrase should live in pair_cast",
|
||||
)
|
||||
partner_styling = pair_cast.softcore_partner_styling(
|
||||
seed_config={},
|
||||
seed=1,
|
||||
row_number=1,
|
||||
women_count=2,
|
||||
men_count=1,
|
||||
pov_labels=["Man A"],
|
||||
label_map={"Woman B": {"softcore_outfit": "custom satin dress"}, "Man A": {"softcore_outfit": "hidden"}},
|
||||
axis_rng=lambda _config, _axis, seed_value, row_value: random.Random(seed_value + row_value),
|
||||
choose=lambda _rng, pool: pool[0],
|
||||
slot_softcore_outfit=lambda slot, _rng: str((slot or {}).get("softcore_outfit") or ""),
|
||||
)
|
||||
_expect(
|
||||
partner_styling["outfits"] == ["Woman B wears custom satin dress"],
|
||||
"Pair cast partner styling should use configured partner outfit and skip POV men",
|
||||
)
|
||||
_expect_text("pair_cast.partner_pose", partner_styling.get("pose"), 12)
|
||||
options = json.loads(
|
||||
pb.build_insta_of_options_json(
|
||||
softcore_expression_enabled="false",
|
||||
|
||||
Reference in New Issue
Block a user