我想在一个数字中找到数字的平方和，我的代码对正数很好，但对负数似乎不起作用

def find_square_sum(num):
    if num >= 0:
        square_sum = sum([i**2 for i in range(num+1)])
        return square_sum
    else:
        return "输入的数字必须是正数或零"

num = int(input("请输入一个数字："))
result = find_square_sum(num)
print("数字的平方和为：", result)