Improv

Author: ShikharJ

symengine/lib/symengine_wrapper.pyx

@@ -1802,7 +1802,7 @@ cdef class DenseMatrixBase(MatrixBase):
         cdef Basic e_
         for i in range(self.rows):
             for j in range(self.cols):
-                e_ = _sympify(self._get(i, j))
+                e_ = self._get(i, j)
                 deref(result.thisptr).set(i, j, e_.thisptr)
         for i in range(o.rows):
             for j in range(o.cols):
@@ -1818,7 +1818,7 @@ cdef class DenseMatrixBase(MatrixBase):
         cdef Basic e_
         for i in range(self.rows):
             for j in range(self.cols):
-                e_ = _sympify(self._get(i, j))
+                e_ = self._get(i, j)
                 deref(result.thisptr).set(i, j, e_.thisptr)
         for i in range(o.rows):
             for j in range(o.cols):
@@ -2128,7 +2128,7 @@ cdef class DenseMatrixBase(MatrixBase):
            return self.free_symbols
 
     def simplify(self, *args, **kwargs):
-        return self.applyfunc(lambda x : x.simplify(*args, **kwargs))
+        return self._applyfunc(lambda x : x.simplify(*args, **kwargs))
 
     def expand(self, *args, **kwargs):
         return self.applyfunc(lambda x : x.expand())
@@ -2166,14 +2166,12 @@ cdef class MutableDenseMatrix(DenseMatrixBase):
         for k in range(0, self.cols):
             self[i, k], self[j, k] = self[j, k], self[i, k]
 
-    def applyfunc(self, f):
-        cdef DenseMatrixBase out = self.__class__(self)
+    def _applyfunc(self, f):
         cdef int nr = self.nrows()
         cdef int nc = self.ncols()
         for i in range(nr):
             for j in range(nc):
-                out._set(i, j, f(self._get(i, j)))
-        return out
+                self._set(i, j, f(self._get(i, j)))
 
 Matrix = DenseMatrix = MutableDenseMatrix