Driver code
if __name__ == " _ _ main _ _ " : NEW_LINE INDENT N = 5 NEW_LINE K = 1 NEW_LINE if ( CheckBits ( N , K ) ) : NEW_LINE INDENT print ( " YES " ) NEW_LINE DEDENT else : NEW_LINE INDENT print ( " NO " ) NEW_LINE DEDENT DEDENT