编写一个函数，该函数的功能是判断一个整数是不是素数，在main函数中输入一个整数，调用该函数。

以下是Python代码实现：

def is_prime(num):
    if num < 2:
        return False
    for i in range(2, int(num ** .5) + 1):
        if num % i == :
            return False
    return True

if __name__ == '__main__':
    num = int(input("请输入一个整数："))
    if is_prime(num):
        print(num, "是素数")
    else:
        print(num, "不是素数")

函数is_prime接受一个整数作为参数，如果该整数小于2，则直接返回False；否则，从2开始到该整数的平方根（向上取整）的范围内遍历，如果该整数能被任何一个数整除，则返回False，否则返回True。

在main函数中，首先输入一个整数，然后调用is_prime函数判断该整数是否为素数，最后输出结果。