Extract penetration role graph wording
This commit is contained in:
@@ -885,6 +885,88 @@ def smoke_pov_oral_position_routes() -> None:
|
||||
_expect(term in prompt, f"{name} Krea prompt missing {term!r}: {prompt}")
|
||||
|
||||
|
||||
def smoke_pov_penetration_position_routes() -> None:
|
||||
cases = [
|
||||
(
|
||||
"pov_penetration_missionary",
|
||||
"missionary",
|
||||
("woman a lies on her back", "man a is above her between her thighs"),
|
||||
("pov missionary position", "viewer is above her", "penetrates her pussy"),
|
||||
),
|
||||
(
|
||||
"pov_penetration_cowgirl",
|
||||
"cowgirl",
|
||||
("woman a straddles man a's hips facing him", "man a lies under her"),
|
||||
("pov cowgirl position", "viewer lies on his back", "woman straddles his hips"),
|
||||
),
|
||||
(
|
||||
"pov_penetration_reverse_cowgirl",
|
||||
"reverse_cowgirl",
|
||||
("woman a straddles man a's hips facing away", "man a lies under her"),
|
||||
("pov reverse cowgirl position", "facing away", "viewer lies on his back"),
|
||||
),
|
||||
(
|
||||
"pov_penetration_doggy",
|
||||
"doggy",
|
||||
("woman a is on all fours", "man a is positioned behind her"),
|
||||
("ass raised toward the pov viewer", "on all fours", "penetrates her pussy"),
|
||||
),
|
||||
(
|
||||
"pov_penetration_edge_supported",
|
||||
"edge_supported",
|
||||
("raised edge", "man a kneels between her thighs"),
|
||||
("pov raised-edge penetration position", "viewer kneels between her legs", "penetrates her pussy"),
|
||||
),
|
||||
(
|
||||
"pov_penetration_lotus",
|
||||
"lotus_lap",
|
||||
("woman a sits in man a's lap", "legs around his hips"),
|
||||
("pov lotus position", "woman sits in his lap", "penetrates her pussy"),
|
||||
),
|
||||
]
|
||||
for offset, (name, position_key, role_terms, krea_terms) in enumerate(cases, start=3801):
|
||||
pair = pb.build_insta_of_pair(
|
||||
row_number=1,
|
||||
start_index=1,
|
||||
seed=offset,
|
||||
ethnicity="any",
|
||||
figure="random",
|
||||
no_plus_women=False,
|
||||
no_black=False,
|
||||
trigger=Trigger,
|
||||
prepend_trigger_to_prompt=True,
|
||||
options_json=_insta_options(
|
||||
softcore_camera_mode="from_camera_config",
|
||||
hardcore_camera_mode="from_camera_config",
|
||||
camera_detail="compact",
|
||||
),
|
||||
character_cast=_character_cast(pov_man=True),
|
||||
hardcore_position_config=_position_filter("penetration_only", "penetrative", [position_key]),
|
||||
location_config=_coworking_location_config(),
|
||||
hardcore_camera_config=_orbit_camera(
|
||||
horizontal_angle=45,
|
||||
vertical_angle=0,
|
||||
zoom=7.5,
|
||||
subject_focus="action",
|
||||
),
|
||||
)
|
||||
_expect_pair(pair, name)
|
||||
hard_row = pair.get("hardcore_row") or {}
|
||||
_expect(hard_row.get("action_family") == "penetration", f"{name} action_family should be penetration")
|
||||
_expect(hard_row.get("position_family") == "penetrative", f"{name} position_family should be penetrative")
|
||||
_expect(position_key in (hard_row.get("position_keys") or []), f"{name} lost position key {position_key!r}")
|
||||
role_graph = _expect_text(f"{name}.source_role_graph", hard_row.get("source_role_graph"), 40).lower()
|
||||
for term in role_terms:
|
||||
_expect(term in role_graph, f"{name} role graph missing {term!r}: {role_graph}")
|
||||
krea = krea_formatter.format_krea2_prompt("", metadata_json=_json(pair), target="hardcore")
|
||||
prompt = _expect_text(f"{name}.krea_prompt", krea.get("krea_prompt"), 60).lower()
|
||||
_expect("metadata" in krea.get("method", ""), f"{name}.krea did not use metadata")
|
||||
_expect("viewer" in prompt and "pov" in prompt, f"{name} Krea prompt lost POV wording")
|
||||
_expect("camera:" not in krea.get("krea_prompt", ""), f"{name} Krea prompt emitted normal third-person camera directive")
|
||||
for term in krea_terms:
|
||||
_expect(term in prompt, f"{name} Krea prompt missing {term!r}: {prompt}")
|
||||
|
||||
|
||||
def smoke_no_expression_fallback() -> None:
|
||||
cast = pb.build_character_slot_json(
|
||||
subject_type="woman",
|
||||
@@ -1018,6 +1100,7 @@ SMOKE_CASES: list[tuple[str, Callable[[], None]]] = [
|
||||
("krea_pov_penetration_route", smoke_krea_pov_penetration_route),
|
||||
("pov_outercourse_position_routes", smoke_pov_outercourse_position_routes),
|
||||
("pov_oral_position_routes", smoke_pov_oral_position_routes),
|
||||
("pov_penetration_position_routes", smoke_pov_penetration_position_routes),
|
||||
("expression_disabled", smoke_no_expression_fallback),
|
||||
("formatter_metadata_fixtures", smoke_formatter_metadata_fixtures),
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user