def test_case5(self):
root = create_tree_from_array([3, 0, 4])
sub = create_tree_from_array([2, 1])
# merge
root.left.right = sub
actual = self.s.trimBST(root, 1, 3)
self.assertEqual(root.val, 3)
self.assertEqual(root.left.val, 2)
self.assertTrue(root.left.right is None)
self.assertEqual(root.left.left.val, 1)
def test_case4(self):
root = my_tree.create_tree_from_array(
[4, 1, 6, 0, 2, 5, 7, None, None, None, 3, None, None, None, 8])
result_root = self.s.bstToGst(root)
self.assertEqual(result_root.val, 30)
self.assertEqual(result_root.left.val, 36)
self.assertEqual(result_root.left.left.val, 36)
def test_case3(self):
root = my_tree.create_tree_from_array([3, 2, 4, 1])
result_root = self.s.bstToGst(root)
self.assertEqual(result_root.val, 7)
self.assertEqual(result_root.right.val, 4)
self.assertEqual(result_root.left.val, 9)
self.assertEqual(result_root.left.left.val, 10)
def test_case1(self):
root = create_tree_from_array([5, 3, 6, 2, 4, None, 7])
actual = self.s.findTarget(root, 9)
self.assertEqual(True, actual)
def test_case4(self):
root = create_tree_from_array([2, 1, 3])
actual = self.s.findTarget(root, 3)
self.assertEqual(True, actual)
def test_case1(self):
root = my_tree.create_tree_from_array([0, None, 1])
result_root = self.s.bstToGst(root)
self.assertEqual(result_root.val, 1)
self.assertEqual(result_root.right.val, 1)
def test_case2(self):
root = my_tree.create_tree_from_array([1, 0, 2])
result_root = self.s.bstToGst(root)
self.assertEqual(result_root.val, 3)
self.assertEqual(result_root.right.val, 2)
self.assertEqual(result_root.left.val, 3)
def test_case_1(self):
s = Solution()
root = create_tree_from_array([1, 1, 1, 1, 1, None, 1])
actual = s.isUnivalTree(root)
self.assertEqual(True, actual)
def test_case_6(self):
s = Solution()
root = create_tree_from_array([1, 1, 1, 2])
actual = s.isUnivalTree(root)
self.assertEqual(False, actual)
def test_case1(self):
head = create_tree_from_array([1, 0, 2])
actual = self.s.trimBST(head, 1, 2)
self.assertTrue(actual.left is None)
def test_case4(self):
head = create_tree_from_array([1, None, 2])
actual = self.s.trimBST(head, 2, 3)
self.assertEqual(actual.val, 2)
def test_case2(self):
head = create_tree_from_array([1])
actual = self.s.trimBST(head, 1, 2)
self.assertTrue(actual.val, 1)
def test_case1(self):
root = my_tree.create_tree_from_array([1, 2, 3, None, 5, None, None])
actual = self.s.binaryTreePaths(root)
self.assertEqual(actual, ["1->2->5", "1->3"])
def test_case4(self):
root = my_tree.create_tree_from_array([1, None, 3])
actual = self.s.binaryTreePaths(root)
self.assertEqual(actual, ["1->3"])