Happy Numbers

1def sum_square_digits(n):
2    total = 0
3    while n > 0:
4        digit = n % 10
5        total += digit * digit
6        n //= 10
7    return total
8
9def is_happy(n):
10    seen = set()
11    while n != 1 and n not in seen:
12        seen.add(n)
13        n = sum_square_digits(n)
14    return n == 1
15
16# Test
17num = int(input("Enter a number: "))
18if is_happy(num):
19    print("Happy number")
20else:
21    print("Not a happy number")