浏览代码

test_polygons_to_mask -> test_shape_to_mask

Kentaro Wada 6 年之前
父节点
当前提交
92b3b9bac9
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 3 3
      tests/test_utils.py

+ 3 - 3
tests/test_utils.py

@@ -67,11 +67,11 @@ def test_shapes_to_label():
     assert cls.shape == img.shape[:2]
 
 
-def test_polygons_to_mask():
+def test_shape_to_mask():
     img, data = _get_img_and_data()
     for shape in data['shapes']:
-        polygons = shape['points']
-        mask = labelme.utils.polygons_to_mask(img.shape[:2], polygons)
+        points = shape['points']
+        mask = labelme.utils.shape_to_mask(img.shape[:2], points)
         assert mask.shape == img.shape[:2]