功能对等

Question

我必须解决一项任务，但我无法解决。

我有函数列表，我必须在列表中找到下一个函数：函数适用于某个整数并返回x ^ 2.

例如：

def f0(x):
    return str(x)

def f1(x):
    return x * x

def f2(x):
    return [x] * x

def f3(x):
    raise AssertionError

functions = [f0, f1, f2, f3]

我的代码：

k = lambda x: x*x
j = 0
n = int(input())
for i in functions:
    if i(n) == k(n):
        j += 1
print(j)

但检查系统会写关于EOFError，但在我的终端中它没问题。如何在不应用循环体的情况下制作它？