def square_digits(num):
    return int(''.join([str(int(x)**2) for x in str(num)]))

test.assert_equals(square_digits(9119), 811181)
test.assert_equals(square_digits(55), 2525)