Algoritmia - programa que a través del uso de una función determine si dos números son amigos

import math

def obtener_divisores(n):

    large_divisors = []

    for i in xrange(1, int(math.sqrt(n) + 1)):

        if n % i == 0:

            yield i

            if i*i != n and i != 1:

                large_divisors.append(n / i)

    for divisor in reversed(large_divisors):

        yield divisor

def son_amigos(a, b):

    a_divisores = obtener_divisores(a)

    if sum(a_divisores) == b:

        b_divisores = obtener_divisores(b)

        return sum(b_divisores) == a

    return False

assert son_amigos(220, 284) is True