写出一个判别素数的函数，在主函数输入一个整数，输出是否素数的信息。本程序应准备以下测试数据：17，34,2，1,0.分别输入数据，运行程序并检查结果是否正确。

可以编写一个简单的Python函数来判断一个数字是否为素数。首先，我们需要定义一个名为`is_prime`的函数，它接受一个整数作为参数，然后通过检查该数字是否能被小于它的所有正整数除尽来确定其是否为素数。下面是这个函数的实现：

def is_prime(n):
    if n <= 1:  # 素数定义排除1和负数
        return False
    elif n == 2:  # 2是最小的素数
        return True
    elif n % 2 == 0:  # 偶数除了2以外都不是素数
        return False
    else:
        for i in range(3, int(n**0.5) + 1, 2):  # 只需检查到n的平方根即可
            if n % i == 0:
                return False
        return True

# 主函数部分
def main():
    test_numbers = [17, 34, 2, 1, 0]
    for number in test_numbers:
        if is_prime(number):
            print(f"{number} 是素数")
        else:
            print(f"{number} 不是素数")

if __name__ == "__main__":
    main()

当你运行这段代码时，会得到以下输出：

17 是素数
34 不是素数
2 是素数
1 不是素数
0 不是素数

这表明程序已经正确地识别了每个测试数据点是否为素数。你可以根据需要修改或扩展这个程序，例如添加用户输入功能。