def test_get(tree):
tree = bst.put(tree, 10, 'book10')
tree = bst.put(tree, 7, 'book7')
tree = bst.put(tree, 30, 'book30')
tree = bst.put(tree, 5, 'book5')
tree = bst.put(tree, 4, 'book4')
tree = bst.put(tree, 3, 'book3')
assert bst.size(tree) == 6
node = bst.get(tree, 3)
assert node['value'] == 'book3'
node = bst.get(tree, 34)
assert node is None
def get(map, key):
"""
Retorna la pareja lleve-valor con llave igual a key
Args:
map: La tabla de simbolos
key: La llave asociada a la pareja
Returns:
La tabla de simbolos con la nueva pareja
Raises:
Exception
"""
if (map['type'] == 'BST'):
return bst.get(map, key)
else:
return rbt.get(map, key)
