Fixed orientation of matrix.sum(axis=1), added unit test

1 files changed, 11 insertions, 0 deletions

diff --git a/numpy/core/defmatrix.py b/numpy/core/defmatrix.py
index d506220a2..9234d2f53 100644
--- a/numpy/core/defmatrix.py
+++ b/numpy/core/defmatrix.py
@@ -188,6 +188,17 @@ class matrix(N.ndarray):
     def __str__(self):
         return str(self.__array__())
 
+    def sum(self, axis=None, dtype=None):
+        """Sum the matrix over the given axis.  If the axis is None, sum
+        over all dimensions.  This preserves the orientation of the
+        result as a row or column.
+        """
+        s = N.ndarray.sum(self, axis, dtype)
+        if axis==1:
+            return s.transpose()
+        else:
+            return s
+    
     # Needed becase tolist method expects a[i] 
     #  to have dimension a.ndim-1
     def tolist(self):