if b <= 0:
    return False
elif not math.sqrt(b) % 1:
    return True
return False
answer = [i for i in range(1, 1001) if isSumOfSquares(i ** 2)]
List of numbers whose squares are the sum of two squares