add assertion & simple unittest

.circleci/config.yml

@@ -22,6 +22,10 @@ jobs:
             cd PythonAPI && python setup.py sdist
             python -m pip install --progress-bar off dist/*.tar.gz
           name: Install From Distribution
+      - run:
+          command: |
+            python tests/test_cases.py
+          name: Run Simple Test Cases
 
 workflows:
   main:

PythonAPI/pycocotools/_mask.pyx

@@ -215,6 +215,7 @@ def iou( dt, gt, pyiscrowd ):
     gt = _preproc(gt)
     m = _len(dt)
     n = _len(gt)
+    assert len(pyiscrowd) == n, "iou(iscrowd=) must have the same length as gt"
     if m == 0 or n == 0:
         return []
     if not type(dt) == type(gt):

tests/test_cases.py

@@ -0,0 +1,25 @@
+import unittest
+import numpy as np
+import pycocotools.mask as mask_util
+
+def _encode(x):
+    return mask_util.encode(np.asfortranarray(x, np.uint8))
+
+class TestToBBox(unittest.TestCase):
+    def testToBboxFullImage(self):
+        mask = np.array([[0, 1], [1, 1]])
+        bbox = mask_util.toBbox(_encode(mask))
+        self.assertTrue(
+            (bbox == np.array([0, 0, 2, 2], dtype="float32")).all(),
+            bbox)
+
+    def testToBboxNonFullImage(self):
+        mask = np.zeros((10, 10), dtype=np.uint8)
+        mask[2:4, 3:6] = 1
+        bbox = mask_util.toBbox(_encode(mask))
+        self.assertTrue(
+            (bbox == np.array([3, 2, 3, 2], dtype="float32")).all(),
+            bbox)
+
+if __name__ == "__main__":
+    unittest.main()