Merge pull request #36 from DeepLabCut/AlexEMG-patch-1

Author: MMathisLab

tests/test_pytorch_models.py

@@ -16,33 +16,33 @@
 
 
 @pytest.mark.parametrize(
-    "data",
+    "detector_data",
     [
         ("superanimal_bird", ["ssdlite"]),
         ("superanimal_topviewmouse", ["fasterrcnn_resnet50_fpn_v2"]),
         ("superanimal_quadruped", ["fasterrcnn_resnet50_fpn_v2"]),
-        ("superanimal_humanbody", ["rtmpose_x"]),
     ]
 )
-def test_get_super_animal_detectors(data: tuple[str, list[str]]):
-    dataset, expected_detectors = data
+def test_get_super_animal_detectors(detector_data: tuple[str, list[str]]):
+    dataset, expected_detectors = detector_data
     detectors = modelzoo.get_available_detectors(dataset)
     assert len(detectors) >= len(expected_detectors)
     for det in expected_detectors:
         assert det in detectors
 
 
 @pytest.mark.parametrize(
-    "data",
+    "posemodel_data",
     [
         ("superanimal_bird", ["resnet_50"]),
         ("superanimal_topviewmouse", ["hrnet_w32"]),
         ("superanimal_quadruped", ["hrnet_w32"]),
+        ("superanimal_humanbody", ["rtmpose_x"]),
     ]
 )
-def test_get_super_animal_pose_models(data: tuple[str, list[str]]):
-    dataset, expected_pose_models = data
+def test_get_super_animal_pose_models(posemodel_data: tuple[str, list[str]]):
+    dataset, expected_pose_models = posemodel_data
     pose_models = modelzoo.get_available_models(dataset)
     assert len(pose_models) >= len(expected_pose_models)
     for pose_model in expected_pose_models:
-        assert pose_model in pose_models
+        assert pose_model in pose_models