def setUp(self): self.solution = Solution() self.tree_root = TreeNode(3) self.tree_root.left = TreeNode(9) self.tree_root.right = TreeNode(20) self.tree_root.right.left = TreeNode(15) self.tree_root.right.right = TreeNode(7)
def test_rightSideView_dfs1(self): node_root = TreeNode(1) node_root.left = TreeNode(2) node_root.left.right = TreeNode(5) right_side_view_list = self.solution.rightSideView_dfs(node_root) self.assertIsNotNone(right_side_view_list) self.assertEqual(3, len(right_side_view_list)) self.assertEqual(1, right_side_view_list[0]) self.assertEqual(2, right_side_view_list[1]) self.assertEqual(5, right_side_view_list[2])
def test_rightSideView_dfs_root(self): node_root = TreeNode(1) right_side_view_list = self.solution.rightSideView_dfs(node_root) self.assertIsNotNone(right_side_view_list) self.assertEqual(1, len(right_side_view_list)) self.assertEqual(1, right_side_view_list[0])
def setUp(self): self.solution = Solution() self.tree_alpha_root_node = TreeNode(2) self.tree_alpha_root_node.left = TreeNode(1) self.tree_alpha_root_node.right = TreeNode(3) self.tree_beta_root_node = TreeNode(5) self.tree_beta_root_node.left = TreeNode(1) self.tree_beta_root_node.right = TreeNode(4) self.tree_beta_root_node.right.left = TreeNode(3) self.tree_beta_root_node.right.right = TreeNode(6) self.tree_maxint_alpha_root_node = TreeNode(maxint - 1) self.tree_maxint_alpha_root_node.left = TreeNode(maxint - 2) self.tree_maxint_alpha_root_node.right = TreeNode(maxint) minint = - maxint - 1 self.tree_minint_alpha_root_node = TreeNode(minint + 1) self.tree_minint_alpha_root_node.left = TreeNode(minint) self.tree_minint_alpha_root_node.right = TreeNode(minint + 2)
def test_rightSideView_dfs_none(self): node_root = TreeNode(1) right_side_view_list = self.solution.rightSideView_dfs(None) self.assertIsNone(right_side_view_list)
def test_lowestCommonAncestorBST(self): node_root = TreeNode(6) node_root.left = TreeNode(2) node_root.left.left = TreeNode(0) node_root.left.right = TreeNode(4) node_root.left.right.left = TreeNode(3) node_root.left.right.right = TreeNode(5) node_root.right = TreeNode(8) node_root.right.left = TreeNode(7) node_root.right.right = TreeNode(9) lca = self.solution.lowestCommonAncestorBST(node_root, node_root.left, node_root.right) self.assertIsNotNone(lca) self.assertEqual(node_root, lca)
def test_maxDepth_instinct_leaf_root(self): root_leaf = TreeNode(0) tree_depth = self.solution.maxDepth_instinct(root_leaf) self.assertEqual(1, tree_depth)
def test_minDepth_recursion_leaf_root(self): leaf_root = TreeNode(1) min_depth = self.solution.minDepth_recursion(leaf_root) self.assertIsNotNone(min_depth) self.assertEqual(1, min_depth)