mirror of
https://github.com/donnemartin/interactive-coding-challenges.git
synced 2024-03-22 13:11:13 +08:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import unittest
|
|
|
|
|
|
class TestLowestCommonAncestor(unittest.TestCase):
|
|
|
|
def test_lca(self):
|
|
node10 = Node(10)
|
|
node5 = Node(5)
|
|
node12 = Node(12)
|
|
node3 = Node(3)
|
|
node1 = Node(1)
|
|
node8 = Node(8)
|
|
node9 = Node(9)
|
|
node18 = Node(18)
|
|
node20 = Node(20)
|
|
node40 = Node(40)
|
|
node3.left = node1
|
|
node3.right = node8
|
|
node5.left = node12
|
|
node5.right = node3
|
|
node20.left = node40
|
|
node9.left = node18
|
|
node9.right = node20
|
|
node10.left = node5
|
|
node10.right = node9
|
|
root = node10
|
|
node0 = Node(0)
|
|
binary_tree = BinaryTree()
|
|
self.assertEqual(binary_tree.lca(root, node0, node5), None)
|
|
self.assertEqual(binary_tree.lca(root, node5, node0), None)
|
|
self.assertEqual(binary_tree.lca(root, node1, node8), node3)
|
|
self.assertEqual(binary_tree.lca(root, node12, node8), node5)
|
|
self.assertEqual(binary_tree.lca(root, node12, node40), node10)
|
|
self.assertEqual(binary_tree.lca(root, node9, node20), node9)
|
|
self.assertEqual(binary_tree.lca(root, node3, node5), node5)
|
|
print('Success: test_lca')
|
|
|
|
|
|
def main():
|
|
test = TestLowestCommonAncestor()
|
|
test.test_lca()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|