add coverage (open-mmlab#100)

Author: jin-s13

tests/test_api/test_inference.py

@@ -0,0 +1,42 @@
+from mmpose.apis import (inference_bottom_up_pose_model,
+                         inference_top_down_pose_model, init_pose_model,
+                         vis_pose_result)
+
+
+def test_top_down_demo():
+    skeleton = [[16, 14], [14, 12], [17, 15], [15, 13], [12, 13], [6, 12],
+                [7, 13], [6, 7], [6, 8], [7, 9], [8, 10], [9, 11], [2, 3],
+                [1, 2], [1, 3], [2, 4], [3, 5], [4, 6], [5, 7]]
+
+    # build the pose model from a config file and a checkpoint file
+    pose_model = init_pose_model(
+        'configs/top_down/resnet/coco/res50_coco_256x192.py',
+        None,
+        device='cpu')
+
+    image_name = 'tests/data/coco/000000000785.jpg'
+    # test a single image, with a list of bboxes.
+    pose_results = inference_top_down_pose_model(
+        pose_model, image_name, [[50, 50, 50, 100]], format='xywh')
+
+    # show the results
+    vis_pose_result(pose_model, image_name, pose_results, skeleton=skeleton)
+
+
+def test_bottom_up_demo():
+    skeleton = [[16, 14], [14, 12], [17, 15], [15, 13], [12, 13], [6, 12],
+                [7, 13], [6, 7], [6, 8], [7, 9], [8, 10], [9, 11], [2, 3],
+                [1, 2], [1, 3], [2, 4], [3, 5], [4, 6], [5, 7]]
+
+    # build the pose model from a config file and a checkpoint file
+    pose_model = init_pose_model(
+        'configs/bottom_up/resnet/coco/res50_coco_512x512.py',
+        None,
+        device='cpu')
+
+    image_name = 'tests/data/coco/000000000785.jpg'
+
+    pose_results = inference_bottom_up_pose_model(pose_model, image_name)
+
+    # show the results
+    vis_pose_result(pose_model, image_name, pose_results, skeleton=skeleton)

tests/test_datasets/test_bottom_up_dataset.py

@@ -37,3 +37,4 @@ def test_bottom_up_COCO_dataset():
         pipeline=[])
 
     assert custom_dataset.num_images == 4
+    _ = custom_dataset[0]

tests/test_datasets/test_top_down_dataset.py

@@ -69,6 +69,7 @@ def test_top_down_COCO_dataset():
     image_id = 785
     assert image_id in custom_dataset.image_set_index
     assert len(custom_dataset.image_set_index) == 4
+    _ = custom_dataset[0]
 
 
 def test_top_down_OneHand10K_dataset():
@@ -114,6 +115,7 @@ def test_top_down_OneHand10K_dataset():
 
     assert custom_dataset.test_mode is False
     assert custom_dataset.num_images == 4
+    _ = custom_dataset[0]
 
 
 def test_top_down_MPII_dataset():
@@ -151,6 +153,7 @@ def test_top_down_MPII_dataset():
         pipeline=[])
 
     assert len(custom_dataset) == 5
+    _ = custom_dataset[0]
 
 
 def test_top_down_MPII_TRB_dataset():
@@ -188,6 +191,7 @@ def test_top_down_MPII_TRB_dataset():
         test_mode=True)
 
     assert custom_dataset.test_mode is True
+    _ = custom_dataset[0]
 
 
 def test_top_down_AIC_dataset():
@@ -252,3 +256,4 @@ def test_top_down_AIC_dataset():
     image_id = 1
     assert image_id in custom_dataset.image_set_index
     assert len(custom_dataset.image_set_index) == 3
+    _ = custom_dataset[0]