Python编程函数使用，素数判断

编写一个函数isprime(n),判断是否为素数，在主程序中测试此函数，输入一个整数，用该函数判断，当为素数时，输出1，否则，输出0

def isprime(n):
    for i in range(2,n):
        if n%i==0:
            return 0
    else:
        return 1
n = int(input())
print(isprime(n))

def isprime(n):
    for i in range(2,int(n**0.5)+1):
        if n%i==0:
            return 0
    return 1
    
n = int(input())
print(isprime(n))