Light more light

light_morelight.py

import math
def isPerfectSquare(x):
    sr = math.sqrt(x)
    return ((sr - math.floor(sr) - 0) == 0)


def main():
         while True:
            inp = int(input())
            if inp == 0:
                return
            if inp == 1:
                print('yes')
                continue
            # if perfect square then number of factors is odd
            if isPerfectSquare(inp):
                print("yes")
            else:
                print("no")


if __name__ == "__main__":
    main()