Improved unit tests

Author: jeylau

tests/test_modeldownload.py

@@ -11,18 +11,25 @@
 import dlclibrary
 import os
 import pytest
+from dlclibrary.dlcmodelzoo.modelzoo_download import MODELOPTIONS
 
 
-def test_download_huggingface_model(tmp_path_factory):
-    folder = tmp_path_factory.mktemp("cat")
-    dlclibrary.download_huggingface_model("full_cat", str(folder))
+def test_download_huggingface_model(tmp_path_factory, model="full_cat"):
+    folder = tmp_path_factory.mktemp("temp")
+    dlclibrary.download_huggingface_model(model, str(folder))
 
     assert os.path.exists(folder / "pose_cfg.yaml")
-    assert os.path.exists(folder / "snapshot-75000.meta")
+    assert any(f.startswith("snapshot-") for f in os.listdir(folder))
     # Verify that the Hugging Face folder was removed
     assert not any(f.startswith("models--") for f in os.listdir(folder))
 
 
 def test_download_huggingface_wrong_model():
     with pytest.raises(ValueError):
         dlclibrary.download_huggingface_model("wrong_model_name")
+
+
+@pytest.mark.skip
+@pytest.mark.parametrize("model", MODELOPTIONS)
+def test_download_all_models(tmp_path_factory, model):
+    test_download_huggingface_model(tmp_path_factory, model)