判断一个整数是否为2的次幂

#判断一个整数是否为2的次幂

def mi(n):
    if( n == 2):
        print('yes')
        return
    elif(n == 0):
        print('no')
        return
    while(n%2 == 0):
        n = n // 2
        if(n == 1):
            print('yes')
            return
    print('no')
    return

if __name__ == '__main__':
    a = input()
    mi(int(a))