写一个判断素数的函数

 def my_math(a):
     if a==1:
         print("该数不是素数")
         return
     elif a==2:
         print("该数是素数")
         return
     else:
         i=1
     while i<=a:
         if a%(i+1)==0:    #判断一个数不是素数
             print("该数不是素数")
             break
         else:
             print("该数是素数")
             break
     i+=1
 my_math(3)