gridspec.GridSpecBase.__getitem__ checks its index (original patch from Paul Ivanov)

leejjoon · leejjoon · commit a9686915fefc · 2011-01-05T15:43:30.000Z
svn path=/trunk/matplotlib/; revision=8888
diff --git a/lib/matplotlib/gridspec.py b/lib/matplotlib/gridspec.py
@@ -145,6 +145,8 @@ def __getitem__(self, key):
             else:
                 if k1 < 0:
                     k1 += nrows
+                if k1 >= nrows or k1 < 0 :
+                    raise IndexError("index out of range")
                 row1, row2 = k1, k1+1
 
 
@@ -153,6 +155,8 @@ def __getitem__(self, key):
             else:
                 if k2 < 0:
                     k2 += ncols
+                if k2 >= ncols or k2 < 0 :
+                    raise IndexError("index out of range")
                 col1, col2 = k2, k2+1
 
 
@@ -167,6 +171,8 @@ def __getitem__(self, key):
             else:
                 if key < 0:
                     key += total
+                if key >= total or key < 0 :
+                    raise IndexError("index out of range")
                 num1, num2 = key, None
 
 
