update tests to load superanimal models

Author: AlexEMG

tests/test_modeldownload.py

@@ -23,6 +23,24 @@ def test_download_huggingface_model(tmp_path_factory, model="full_cat"):
     # Verify that the Hugging Face folder was removed
     assert not any(f.startswith("models--") for f in os.listdir(folder))
 
+def test_download_huggingface_superanimalmousemodel(tmp_path_factory, model="superanimal_topviewmouse"):
+    folder = tmp_path_factory.mktemp("temp")
+    dlclibrary.download_huggingface_model(model, str(folder))
+
+    assert os.path.exists(folder / "pose_cfg.yaml")
+    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_superquadrupedmodel(tmp_path_factory, model="superanimal_quadruped"):
+    folder = tmp_path_factory.mktemp("temp")
+    dlclibrary.download_huggingface_model(model, str(folder))
+
+    assert os.path.exists(folder / "pose_cfg.yaml")
+    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):
@@ -38,4 +56,5 @@ def test_parse_superanimal_models():
 @pytest.mark.skip
 @pytest.mark.parametrize("model", MODELOPTIONS)
 def test_download_all_models(tmp_path_factory, model):
+    print("Downloading ...", model)
     test_download_huggingface_model(tmp_path_factory, model)

tools/README.md

@@ -13,8 +13,8 @@ from the repository root (this needs `pip install licenseheaders`)
 You can edit the `NOTICE.yml` to update the header.
 
 
-## Running tests
+## Running tests [with more details](https://docs.pytest.org/en/latest/how-to/capture-stdout-stderr.html)
 
 ``` bash
-pytest
+pytest -s
 ```