Fix diff for matrices

jiggzson · Mar 22, 2020 · 5e49cdc · 5e49cdc
1 parent 38e80f7
commit 5e49cdc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/Calculus.js b/Calculus.js
@@ -368,10 +368,17 @@ if((typeof module) !== 'undefined' && typeof nerdamer === 'undefined') {
             if(core.Utils.isVector(symbol)) {
                 var vector = new core.Vector([]);
                 symbol.each(function(x) {
-                    vector.elements.push(__.diff(x, wrt));
+                    vector.elements.push(__.diff(x, wrt, nth));
                 });
                 return vector;
             }
+            else if(core.Utils.isMatrix(symbol)) {
+                var matrix = new core.Matrix();
+                symbol.each(function(x, j, i) {
+                    matrix.set(i, j, __.diff(x, wrt, nth));
+                });
+                return matrix;
+            }
 
             var d = isSymbol(wrt) ? wrt.text() : wrt; 
             //the nth derivative

diff --git a/all.js b/all.js
@@ -13,4 +13,4 @@ require('./Solve.js');
 require('./Extra.js');
 
 //export nerdamer
-module.exports = nerdamer;
+module.exports = nerdamer;