Raise IndexError if matrix index out of bounds

skirpichev · skirpichev · commit a9637115e524 · 2023-04-20T06:21:21.000+03:00
Closes #318
diff --git a/mpmath/matrices/matrices.py b/mpmath/matrices/matrices.py
@@ -454,6 +454,8 @@ def __getitem__(self, key):
                     raise IndexError('Row index out of bounds')
             else:
                 # Single row
+                if key[0] >= self.__rows:
+                    raise IndexError('Row index out of bounds')
                 rows = [key[0]]
 
             # Columns
@@ -468,6 +470,8 @@ def __getitem__(self, key):
 
             else:
                 # Single column
+                if key[1] >= self.__cols:
+                    raise IndexError('Column index out of bounds')
                 columns = [key[1]]
 
             # Create matrix slice
diff --git a/mpmath/tests/test_matrices.py b/mpmath/tests/test_matrices.py
@@ -49,6 +49,8 @@ def test_matrix_basic():
     assert A9 != A10
     assert nstr(A9)
     assert A9 != None  # issue 283
+    pytest.raises(IndexError, lambda: zeros(1,1)[:, 1])  # issue 318
+    pytest.raises(IndexError, lambda: zeros(1,1)[1, :])
 
 def test_matmul():
     """
