BUG: Implement bounds check in chunk getter

NonVolatileComputing · Oct 24, 2017 · bda7f4c · bda7f4c
1 parent 4eb38a2
commit bda7f4c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/pyarrow/table.pxi b/python/pyarrow/table.pxi
@@ -102,6 +102,10 @@ cdef class ChunkedArray:
         pyarrow.Array
         """
         self._check_nullptr()
+
+        if i >= self.num_chunks or i < 0:
+            raise IndexError('Chunk index out of range.')
+
         return pyarrow_wrap_array(self.chunked_array.chunk(i))
 
     def iterchunks(self):