多行输入两个正方形边长，求面积之差是否是质数（素数）

限定行数，输入两个正方形边长，求面积之差是否是质数（素数）：

from numpy import *
while True:
    try:
        Q = int(input())  # 先输入一个数字代表需要输入几行，比如Q=4，那么就需要再输入4行数据
        for i in range(Q):
            a,b = [int(n) for n in input().split()]
            s = a**2-b**2

            if s<2:
                print("这个数不是素数！")
            else:
                for i in range(2,s):
                    if s%i==0:
                        print("no")
                        break
                else:
                    print("yes")
    except:
        break

结果：

2

9 3
no
4 3
yes