16 lines
570 B
Python
16 lines
570 B
Python
from services.hwfit.image_models import rank_image_models, IMAGE_MODEL_REGISTRY
|
|
|
|
SYS = {"gpu_vram_gb": 0, "has_gpu": False}
|
|
|
|
|
|
def test_rank_image_models_handles_non_string_search():
|
|
# search is a CLI/API filter arg; a non-string made search.lower() raise
|
|
# AttributeError. A non-string search should behave as "no filter".
|
|
out = rank_image_models(SYS, search=123)
|
|
assert len(out) == len(IMAGE_MODEL_REGISTRY)
|
|
|
|
|
|
def test_rank_image_models_string_filter_still_applies():
|
|
out = rank_image_models(SYS, search="zzzznotarealmodelzzz")
|
|
assert out == []
|