add test to invert_binary_tree

ronaldngounou · ronaldngounou · commit 7d7e68ca5f86 · 2024-10-16T00:04:39.000-04:00
diff --git a/data_structures/binary_tree/invert_binary_tree.py b/data_structures/binary_tree/invert_binary_tree.py
@@ -41,6 +41,10 @@ def mirror_binary_tree(self, root : TreeNode):
         0
         20
         10
+        >>> tree2 = TreeNode(9)
+        >>> result_tree2 = MirrorBinaryTree().mirror_binary_tree(tree2)
+        >>> print_preorder(result_tree2)
+        9
         """ 
         
         if not root:

Original file line number	Diff line number	Diff line change
`@@ -41,6 +41,10 @@ def mirror_binary_tree(self, root : TreeNode):`
`41`	`41`	`0`
`42`	`42`	`20`
`43`	`43`	`10`
	`44`	`+ >>> tree2 = TreeNode(9)`
	`45`	`+ >>> result_tree2 = MirrorBinaryTree().mirror_binary_tree(tree2)`
	`46`	`+ >>> print_preorder(result_tree2)`
	`47`	`+ 9`
`44`	`48`	`"""`
`45`	`49`
`46`	`50`	`if not root:`