ingresa un numero de 4 cifras que dé como resultado la suma de sus cifras

def sum_of_digits(a):
  if len(str(a)) != 4:
    print "Error" 
  else:
    suma = 0
    for i in str(a):
      suma += int(i)
    return suma
    
sum_of_digits(1234)