def testHeightV2(self): node_2 = Node(2, None, None) node_7 = Node(7, None, None) node_4 = Node(4, node_2, node_7) node_6 = Node(6, None, None) node_5 = Node(5, node_4, node_6) bst = BST(node_5) self.assertEquals(2, bst.get_height(node_5))
def testHeightOneNode(self): node_5 = Node(5, None, None) bst = BST(node_5) self.assertEquals(0, bst.get_height(node_5))