def adata_parameterized(request):
    return gen_adata(shape=(200, 300), X_type=request.param)
def remove(root, key):
    left, right = split(root, key)
    return merge(left, split(right, key + 1)[1])