10 lines
502 B
Python
10 lines
502 B
Python
from services.hwfit.image_models import rank_image_models, IMAGE_MODEL_REGISTRY
|
|
|
|
|
|
def test_rank_image_models_handles_non_dict_system():
|
|
# `system` is the detected-hardware dict; if detection failed and returned
|
|
# None (or a non-dict), system.get(...) raised AttributeError. Treat a
|
|
# non-dict system as "unknown hardware" (no GPU) rather than crashing.
|
|
assert len(rank_image_models(None)) == len(IMAGE_MODEL_REGISTRY)
|
|
assert len(rank_image_models(123)) == len(IMAGE_MODEL_REGISTRY)
|