40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
from mjr_am_backend.features.index.search_hydration import hydrate_asset_payload
|
|
|
|
|
|
def test_hydration_prefers_geninfo_positive_over_stale_positive_prompt():
|
|
asset = {
|
|
"metadata_raw": {
|
|
"positive_prompt": r"C:\ComfyUI\output\image_00001.png",
|
|
"geninfo": {"positive": {"value": "cinematic portrait"}},
|
|
}
|
|
}
|
|
|
|
hydrated = hydrate_asset_payload(asset)
|
|
|
|
assert hydrated["positive_prompt"] == "cinematic portrait"
|
|
|
|
|
|
def test_hydrate_asset_payload_restores_generation_time_from_metadata_raw():
|
|
asset = hydrate_asset_payload(
|
|
{
|
|
"id": 42,
|
|
"filename": "sample.png",
|
|
"metadata_raw": {"generation_time_ms": 8421},
|
|
}
|
|
)
|
|
|
|
assert asset["generation_time_ms"] == 8421
|
|
|
|
|
|
def test_hydrate_asset_payload_preserves_selected_generation_time():
|
|
asset = hydrate_asset_payload(
|
|
{
|
|
"id": 43,
|
|
"filename": "sample.png",
|
|
"generation_time_ms": 9100,
|
|
"metadata_raw": {"generation_time_ms": 8421},
|
|
}
|
|
)
|
|
|
|
assert asset["generation_time_ms"] == 9100
|