[数据结构与算法]相邻的素数python

#输入一个大于2的素数。输出与这个素数相邻的2个素数

import math
def f(x)://判断是否是素数
    n=1
    if x==2:
        return True
    for i in range(2,x):
        if x%i==0:
            n=0
    if n==1:
        return True
    else:
        return False
str=input()//输入一个数
while str!="":
    str=int(str)
    a=0
    b=0
    a1=0
    b1=0
    each1=str-1
    each2=str+1
    while a1==0: //遍历比这个数小的数
        if f(each1): //找到素数
            a=each1
            a1=1 //跳出循环
        each1-=1
    while b1==0://遍历比这个数大的数
        if f(each2): //找到素数
            b=each2
            b1=1 //跳出循环
        each2+=1
    print("%d<%d<%d"%(a,str,b))
    str=input()