编写函数，判断一个数字是否为素数，是则返回Yes，否则返回No。再编写测试函数

以下是判断素数的函数实现：

def is_prime(n):
    """
    判断一个数字是否为素数
    """
    if n < 2:  # 小于2的数字不是素数
        return "No"
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return "No"
    return "Yes"

以下是测试函数的实现：

def test_is_prime():
    """
    is_prime函数的测试函数
    """
    assert is_prime(0) == "No"
    assert is_prime(1) == "No"
    assert is_prime(2) == "Yes"
    assert is_prime(3) == "Yes"
    assert is_prime(4) == "No"
    assert is_prime(5) == "Yes"
    assert is_prime(6) == "No"
    assert is_prime(7) == "Yes"
    assert is_prime(8) == "No"
    assert is_prime(9) == "No"
    assert is_prime(10) == "No"
    assert is_prime(11) == "Yes"
    print("所有测试通过！")

可以使用 `test_is_prime()` 函数来运行测试。