Test partial inversed trees in day 24 challenge

Author: aanton

src/day24/index.test.js

@@ -31,6 +31,62 @@ describe('Day 24 challenge', () => {
     right: {value: 5, left: {value: 4, left: null, right: null}, right: null},
   }
 
+  const treeC = {
+    value: 1,
+    left: {
+      value: 2,
+      left: {value: 3, left: null, right: null},
+      right: {value: 4, left: null, right: null},
+    },
+    right: {
+      value: 5,
+      left: {value: 6, left: null, right: null},
+      right: {value: 7, left: null, right: null},
+    },
+  }
+
+  const treeCInverse = {
+    value: 1,
+    left: {
+      value: 5,
+      left: {value: 7, left: null, right: null},
+      right: {value: 6, left: null, right: null},
+    },
+    right: {
+      value: 2,
+      left: {value: 4, left: null, right: null},
+      right: {value: 3, left: null, right: null},
+    },
+  }
+
+  const treeCPartialInverseTopBranches = {
+    value: 1,
+    left: {
+      value: 5,
+      left: {value: 3, left: null, right: null},
+      right: {value: 4, left: null, right: null},
+    },
+    right: {
+      value: 2,
+      left: {value: 6, left: null, right: null},
+      right: {value: 7, left: null, right: null},
+    },
+  }
+
+  const treeCPartialInverseBottomBranches = {
+    value: 1,
+    left: {
+      value: 2,
+      left: {value: 3, left: null, right: null},
+      right: {value: 4, left: null, right: null},
+    },
+    right: {
+      value: 5,
+      left: {value: 7, left: null, right: null},
+      right: {value: 6, left: null, right: null},
+    },
+  }
+
   test('returns true when comparing the same tree', () => {
     expect(checkIsSameTree(treeA, treeA)).toBeTruthy()
   })
@@ -50,4 +106,18 @@ describe('Day 24 challenge', () => {
   test('returns false when comparing a tree & its partial inverse version (only an inner branch)', () => {
     expect(checkIsSameTree(treeB, treeBPartialInverse)).toBeFalsy()
   })
+
+  test('returns true when comparing a balanced tree & its inverse version', () => {
+    expect(checkIsSameTree(treeC, treeCInverse)).toBeTruthy()
+  })
+
+  test('returns false when comparing a balanced tree & its partial inverse version (top branches)', () => {
+    expect(checkIsSameTree(treeC, treeCPartialInverseTopBranches)).toBeFalsy()
+  })
+
+  test('returns false when comparing a balanced tree & its partial inverse version (bottom branches)', () => {
+    expect(
+      checkIsSameTree(treeC, treeCPartialInverseBottomBranches)
+    ).toBeFalsy()
+  })
 })