TST: Add bounds check tests for chunk getter

NonVolatileComputing · Oct 24, 2017 · cabdd43 · cabdd43
1 parent bda7f4c
commit cabdd43
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/python/pyarrow/tests/test_table.py b/python/pyarrow/tests/test_table.py
@@ -211,6 +211,12 @@ def test_table_basics():
         for chunk in col.data.iterchunks():
             assert chunk is not None
 
+        with pytest.raises(IndexError):
+            col.data.chunk(-1)
+
+        with pytest.raises(IndexError):
+            col.data.chunk(col.data.num_chunks)
+
 
 def test_table_from_arrays_invalid_names():
     data = [