mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-21 21:22:11 -03:00
c12aefa82a
- Use modelVersionId as fallback for all loras in fingerprint calculation (not just deleted) - Add URL-based duplicate detection using source_path field - Combine both fingerprint and URL-based duplicate detection in API response - Fix _download_remote_media return type and unbound variable issue
101 lines
2.8 KiB
Python
101 lines
2.8 KiB
Python
"""Test for modelVersionId fallback in fingerprint calculation."""
|
|
import pytest
|
|
from py.utils.utils import calculate_recipe_fingerprint
|
|
|
|
|
|
def test_calculate_fingerprint_with_model_version_id_fallback():
|
|
"""Test that fingerprint uses modelVersionId when hash is empty, even when not deleted."""
|
|
loras = [
|
|
{
|
|
"hash": "",
|
|
"strength": 1.0,
|
|
"modelVersionId": 2639467,
|
|
"isDeleted": False,
|
|
"exclude": False
|
|
}
|
|
]
|
|
fingerprint = calculate_recipe_fingerprint(loras)
|
|
assert fingerprint == "2639467:1.0"
|
|
|
|
|
|
def test_calculate_fingerprint_with_multiple_model_version_ids():
|
|
"""Test fingerprint with multiple loras using modelVersionId fallback."""
|
|
loras = [
|
|
{
|
|
"hash": "",
|
|
"strength": 1.0,
|
|
"modelVersionId": 2639467,
|
|
"isDeleted": False,
|
|
"exclude": False
|
|
},
|
|
{
|
|
"hash": "",
|
|
"strength": 0.8,
|
|
"modelVersionId": 1234567,
|
|
"isDeleted": False,
|
|
"exclude": False
|
|
}
|
|
]
|
|
fingerprint = calculate_recipe_fingerprint(loras)
|
|
assert fingerprint == "1234567:0.8|2639467:1.0"
|
|
|
|
|
|
def test_calculate_fingerprint_with_deleted_lora():
|
|
"""Test that deleted loras with modelVersionId are still included."""
|
|
loras = [
|
|
{
|
|
"hash": "",
|
|
"strength": 1.0,
|
|
"modelVersionId": 2639467,
|
|
"isDeleted": True,
|
|
"exclude": False
|
|
}
|
|
]
|
|
fingerprint = calculate_recipe_fingerprint(loras)
|
|
assert fingerprint == "2639467:1.0"
|
|
|
|
|
|
def test_calculate_fingerprint_with_excluded_lora():
|
|
"""Test that excluded loras are skipped even with modelVersionId."""
|
|
loras = [
|
|
{
|
|
"hash": "",
|
|
"strength": 1.0,
|
|
"modelVersionId": 2639467,
|
|
"isDeleted": False,
|
|
"exclude": True
|
|
}
|
|
]
|
|
fingerprint = calculate_recipe_fingerprint(loras)
|
|
assert fingerprint == ""
|
|
|
|
|
|
def test_calculate_fingerprint_prefers_hash_over_version_id():
|
|
"""Test that hash is used even when modelVersionId is present."""
|
|
loras = [
|
|
{
|
|
"hash": "abc123",
|
|
"strength": 1.0,
|
|
"modelVersionId": 2639467,
|
|
"isDeleted": False,
|
|
"exclude": False
|
|
}
|
|
]
|
|
fingerprint = calculate_recipe_fingerprint(loras)
|
|
assert fingerprint == "abc123:1.0"
|
|
|
|
|
|
def test_calculate_fingerprint_without_hash_or_version_id():
|
|
"""Test that loras without hash or modelVersionId are skipped."""
|
|
loras = [
|
|
{
|
|
"hash": "",
|
|
"strength": 1.0,
|
|
"modelVersionId": 0,
|
|
"isDeleted": False,
|
|
"exclude": False
|
|
}
|
|
]
|
|
fingerprint = calculate_recipe_fingerprint(loras)
|
|
assert fingerprint == ""
|