Python - MayorSumaDigitos(n1, n2)

#!/usr/bin/python3

import unittest

#suma los digitos de un numero

def suma_digitos(numero):

    suma = 0

    while numero:

        numero, digito = divmod(numero, 10)

        suma += digito

    return suma

#numero cuya suma de digitos es mayor

def mayor_suma_digitos(numero1, numero2):

    suma1 = suma_digitos(numero1)

    suma2 = suma_digitos(numero2)

    if suma1 == suma2:

        return 0

    elif suma1 > suma2:

        return numero1

    else:

        return numero2

class MyTest(unittest.TestCase):

    def test(self):

        self.assertEqual(mayor_suma_digitos(19,91) ,0)

        self.assertEqual(mayor_suma_digitos(191,185) ,185)

        self.assertEqual(mayor_suma_digitos(2250,541) ,541)

if __name__ == '__main__':

    myTest=MyTest()

    myTest.test()

def mayorSumaDigitos(n1, n2):

    s1, s2 = sum(map(int, str(n1))), sum(map(int, str(n2)))

    return (s1 > s2 and n1) or (s1 < s2 and n1) or 0

print(mayorSumaDigitos(31, 21)) # mostrara 31

def mayorSumaDigitos(n1, n2):

    sum1 = sum2 = 0

    for n in str(n1):

        sum1 += int(n)

    for n in str(n2):

        sum2 += int(n)

    if sum1 > sum2:

        return n1

    elif sum1 < sum2:

        return n2

    else:

        return 0

print mayorSumaDigitos(23, 23) # mostrara 0