定义一个函数，输入int类型范围内的一个非负整数，求该整数的各数位上数字的平方和。 并调用该函数

以下是函数的定义和调用示例：

def digit_square_sum(n: int) -> int:
    """
    计算一个非负整数的各数位上数字的平方和
    """
    res = 0
    while n > 0:
        res += (n % 10) ** 2
        n //= 10
    return res

# 调用示例
n = 12345
print(digit_square_sum(n))  # 输出 55

在上述代码中，我们定义了 `digit_square_sum` 函数，它接受一个非负整数 `n` 作为输入，并返回该整数的各数位上数字的平方和。

在函数内部，我们使用了一个循环来计算各位数字的平方和。循环的条件是 `n > 0`，即当 `n` 不为零时，我们就取出 `n` 的个位数，计算其平方后加到结果 `res` 中，并将 `n` 除以 10 将个位数去掉。最后，函数返回结果 `res`。

在调用示例中，我们定义了一个整数 `n`，并传递给 `digit_square_sum` 函数。函数返回该整数的各位数字的平方和，并将结果打印到控制台上。