class TestLC99(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(TestLC99, self).__init__(*args, **kwargs)
self.solution = Solution()
self.root1 = BinaryTree([1, 3, "null", "null", 2]).root
self.root2 = BinaryTree([3, 1, "null", "null", 2]).root
self.bst = BinaryTree([1])
@classmethod
def setUpClass(cls) -> None:
print("set up class...")
@classmethod
def tearDownClass(cls) -> None:
print("tear down class...")
def setUp(self) -> None:
print("-- setup --")
def tearDown(self) -> None:
print("-- tear down --")
def test_recoverTree(self):
self.solution.recoverTree(self.root1)
self.assertTrue(self.bst.is_equal_trees(self.root1, self.root2))
class TestBinaryTree(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(TestBinaryTree, self).__init__(*args, **kwargs)
self.binary_tree = BinaryTree([1, 2, 3, "null", 4, 5, "null"])
@classmethod
def setUpClass(cls) -> None:
print("set up class...")
@classmethod
def tearDownClass(cls) -> None:
print("tear down class...")
def setUp(self) -> None:
print("set up")
def tearDown(self) -> None:
print("tear down")
def test_BinaryTree(self):
self.assertIsNotNone(self.binary_tree)
def test_pre_order_traverse(self):
self.assertEqual([1, 2, 4, 3, 5],
self.binary_tree.pre_order_traverse())
def test_in_order_traverse(self):
self.assertEqual([2, 4, 1, 5, 3], self.binary_tree.in_order_traverse())
def test_after_order_traverse(self):
self.assertEqual([4, 2, 5, 3, 1],
self.binary_tree.after_order_traverse())
def test_build_binary_tree_from_an_incomplete_val_sequence(self):
self.assertIsNotNone(
self.binary_tree.build_binary_tree_from_an_incomplete_val_sequence(
[1, "null", 2, 3]))
def __init__(self, *args, **kwargs):
super(TestLC101, self).__init__(*args, **kwargs)
self.solution = Solution()
self.root = BinaryTree([1, 2, 2, 3, 4, 4, 3]).root
self.root1 = BinaryTree([1, 2, 2, 2, "null", 2]).root
def __init__(self, *args, **kwargs):
super(TestLC100, self).__init__(*args, **kwargs)
self.solution = Solution3()
self.root1 = BinaryTree([1, 2, 3]).root
self.root2 = BinaryTree([1, 2, 3]).root
def __init__(self, *args, **kwargs):
super(TestLC102, self).__init__(*args, **kwargs)
self.solution = Solution()
self.root = BinaryTree([3, 9, 20, "null", "null", 15, 7]).root
def __init__(self, *args, **kwargs):
super(TestLC99, self).__init__(*args, **kwargs)
self.solution = Solution()
self.root1 = BinaryTree([1, 3, "null", "null", 2]).root
self.root2 = BinaryTree([3, 1, "null", "null", 2]).root
self.bst = BinaryTree([1])
def __init__(self, *args, **kwargs):
super(TestLC199, self).__init__(*args, **kwargs)
self.solution = Solution()
self.node = BinaryTree([1, 2, 3, 4, 'null', 'null', 'null', 5]).root
def __init__(self, *args, **kwargs):
super(TestLC98, self).__init__(*args, **kwargs)
self.btree1 = BinaryTree([2, 1, 3]).root
self.btree2 = BinaryTree([5, 1, 4, "null", "null", 3, 6]).root
self.btree3 = BinaryTree([10, 5, 15, "null", "null", 6, 20]).root
self.solution = Solution()
def __init__(self, *args, **kwargs):
super(TestLC94, self).__init__(*args, **kwargs)
self.solution = Solution()
self.binary_tree = BinaryTree([1, "null", 2, 3]).root
def __init__(self, *args, **kwargs):
super(TestBinaryTree, self).__init__(*args, **kwargs)
self.binary_tree = BinaryTree([1, 2, 3, "null", 4, 5, "null"])