Merge pull request sirfz#41 from PeterZhizhin/master

Added segfault test
htcvtc59 · May 9, 2017 · dcf3554 · dcf3554
2 parents 30dbb10 + 79c5934
commit dcf3554
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -163,3 +163,21 @@ def test_end(self):
         self._api.End()
         self._api.SetImageFile(self._image_file)
         self.assertRaises(RuntimeError, self._api.GetUTF8Text)
+
+    @unittest.skipIf(not pil_installed, "Pillow not installed")
+    def test_empty_getcomponents(self):
+        self._api.Init()
+        image = Image.new("RGB", (100, 100), (1, 1, 1))
+        self._api.SetImage(image)
+        result = self._api.GetComponentImages(tesserocr.RIL.TEXTLINE, True)
+        # Test if empty
+        self.assertFalse(result)
+
+    @unittest.skipIf(not pil_installed, "Pillow not installed")
+    def test_empty_small_getcomponents(self):
+        self._api.Init()
+        image = Image.new("RGB", (1, 1), (1, 1, 1))
+        self._api.SetImage(image)
+        result = self._api.GetComponentImages(tesserocr.RIL.TEXTLINE, True)
+        # Test if empty
+        self.assertFalse(result)