A=int(input())
my_list=[]
for i in range(A):
square=int(input())
if 0<square<=1000:
a_values=[]
b_values=[]
tup_list=[]
for x in range(int(square**0.5)+1):
a_values.append(x)
b_values.append(x)
for a in a_values:
for b in b_values:
if (a**2+b**2)==square and a<=b:
tup_list.append((a,b))
my_list.append(tup_list)
for i in my_list:
if len(i) >1:
print(*i,sep=' ')
if len(i)==1:
print(i[0])
My compilation are successful but test cases fails
What to do?