可是下面各位提供的條件我都已經有了阿，不知道問題在哪

可是下面各位提供的條件我都已經有了阿，不知道問題在哪

import math

def sqr(n):

    a=math.sqrt(n)

    if a*a==n:

        return True

a=int(input())

b=[]

for i in range(a):

    s=input().split(' ')

    b.append(s)

for i in b:

    ans=int(i[1])**2-4*int(i[0])*int(i[2])

    if ans<0:

        print('No')

    elif ans>=0 and sqr(ans):

        print('Yes')

    else:

        print('No')

噗，如果不用math模組來判斷就可以過，我過了!!!!!