判断质数

def solve():
    n = 17
    if n < 2:
        print('非质数')
    else:
        is_prime = True
        for i in range(2, int(n**0.5) + 1):
            if n % i == 0:
                is_prime = False
                break
        print('质数' if is_prime else '非质数')