python实现判断一个数是不是素数

本来这是件简单的事，但由于自己的编程基础弱。所以最后的else一直不知道怎样返回false或者True。最后想到while循环，所以把它写下来。
def is_prime(x):
    if x == 0 or x == 1:
        return False
    elif x == 2:
        return True
    else:
        for i in range(2,x+1):
            while i == x:
                return True
                break
            if x % i == 0:
                return False
                break
            
print is_prime(13)