Merge branch 'davidrunger-patch-3'

kolodny · Oct 18, 2015 · 2a729ab · 2a729ab
2 parents 74b4f40 + d3fbd30
commit 2a729ab
Showing 1 changed file with 42 additions and 1 deletion.
diff --git a/invert-tree/test.js b/invert-tree/test.js
@@ -3,7 +3,6 @@ var invertTree = require('./');
 
 describe('invert-tree', function() {
   it('inverts a binary tree', function() {
-
     var root = {value: 6};
     var left = {value: 4};
     var right = {value: 8};
@@ -41,7 +40,49 @@ describe('invert-tree', function() {
         }
       }
     });
+  });
+
+  it('inverts a right-leaning asymmetric tree', function() {
+    var root = {value: 4};
+    var right = {value: 6};
+    var rightOfRight = {value: 8};
+    root.right = right;
+    right.right = rightOfRight;
+
+    invertTree(root);
 
+    root = JSON.parse(JSON.stringify(root));
 
+    assert.deepEqual(root, {
+      "value": 4,
+      "left": {
+        "value": 6,
+        "left": {
+          value: 8
+        }
+      }
+    });
+  });
+
+  it('inverts a left-leaning asymmetric tree', function() {
+    var root = {value: 5};
+    var left = {value: 7};
+    var leftOfLeft = {value: 9};
+    root.left = left;
+    left.left = leftOfLeft;
+
+    invertTree(root);
+
+    root = JSON.parse(JSON.stringify(root));
+
+    assert.deepEqual(root, {
+      "value": 5,
+      "right": {
+        "value": 7,
+        "right": {
+          value: 9
+        }
+      }
+    });
   });
 });