from red_black_tree import RedBlackTree if __name__ == "__main__": items = [3, 4, 10, 1, 6, 5] # items = ['T', 'P', 'R', 'N', 'H', 'O', 'A', 'E', 'I', 'G'] symbol_table = RedBlackTree() for item in items: if symbol_table.contains(item): symbol_table.put(item, symbol_table.get(item) + 1) else: symbol_table.put(item, 1) # red_black_tree.put(item, item) print "BEFORE: ", print symbol_table.keys()