def __init__(self, data=None): if data is None: data = [] self.heap_size = len(data) self._root = BinaryNode.from_array(data) # type: BinaryNode self.heapify()
def test_question_1(self): tree_1 = BinaryNode.from_array([1, 2 ,3, 4, 5, 6, 7]) assert tree_1.is_balanced() leaf_node = next(tree_1.get_leaf_nodes()) leaf_node.rnode = BinaryNode(0) leaf_node.rnode.rnode = BinaryNode(0) assert not tree_1.is_balanced()