Allow empty SA-IS input.

eerimoq · Feb 22, 2019 · 640576a · 640576a
1 parent f67bc65
commit 640576a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/detools/sais.c b/detools/sais.c
@@ -815,13 +815,7 @@ static PyObject* sais_wrapper(PyObject* self_p, PyObject* arg_p)
         return (NULL);
     }
 
-    if (size == 0) {
-        PyErr_SetString(PyExc_ValueError, "SA-IS data empty.");
-
-        return (NULL);
-    }
-
-    suffix_array_p = PyMem_Malloc(size * sizeof(int));
+    suffix_array_p = PyMem_Malloc((size + 1) * sizeof(int));
 
     if (suffix_array_p == NULL) {
         return (NULL);

diff --git a/tests/test_sais.py b/tests/test_sais.py
@@ -13,6 +13,14 @@ class DetoolsSaisTest(unittest.TestCase):
 
     def test_sais(self):
         datas = [
+            (
+                b'',
+                [0]
+            ),
+            (
+                b'1',
+                [1, 0]
+            ),
             (
                 b'1234',
                 [4, 0, 1, 2, 3]