def squaring(f): test = [int(rand() * 1000) for i in range(100)] answer = [x * x for x in test] user_answer = [f(x) for x in test] if (answer == user_answer): Wrapper.printCorrect() else: Wrapper.printWrong(user_answer, answer)