From 24a8d329deff1e63c40460fe41f72dc420ade1c9 Mon Sep 17 00:00:00 2001
From: Travis Oliphant <oliphant@enthought.com>
Date: Fri, 24 Feb 2006 10:48:35 +0000
Subject: Don't allow matrices larger than ndim=2

---
 numpy/core/defmatrix.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'numpy/core/defmatrix.py')

diff --git a/numpy/core/defmatrix.py b/numpy/core/defmatrix.py
index 9234d2f53..65339793a 100644
--- a/numpy/core/defmatrix.py
+++ b/numpy/core/defmatrix.py
@@ -101,6 +101,13 @@ class matrix(N.ndarray):
 
     def __array_finalize__(self, obj):
         ndim = self.ndim
+        if (ndim > 2):
+            newshape = tuple([x for x in self.shape if x > 1])
+            ndim = len(newshape)
+            if ndim == 2:
+                self.shape = newshape
+            elif (ndim > 2):
+                raise ValueError, "shape too large to be a matrix."
         if ndim == 0:
             self.shape = (1,1)
         elif ndim == 1:
-- 
cgit v1.2.1