使用assert语句调试Python代码

当我们编写代码时，我们会经常遇到需要进行调试的情况。在 Python 中，可以使用 assert 语句来进行调试。当一个 assert 语句的条件为 False 时，程序会抛出一个 AssertionError 异常。

下面是一个简单的例子，假设我们要写一个函数来检查一个字符串是否为“pidancode.com”：

def check_string(s):
    assert s == 'pidancode.com', '字符串不是"pidancode.com"'
    print('字符串正确')

在这个例子中，我们使用了 assert 语句来判断输入的字符串是否为“pidancode.com”。如果条件不成立，即输入的字符串不是“pidancode.com”，那么程序会抛出一个 AssertionError 异常，并打印出我们指定的错误信息。

让我们来测试一下这个函数：

check_string('pidancode.com')
# Output: 字符串正确

check_string('皮蛋编程')
# Output: AssertionError: 字符串不是"pidancode.com"

在第一个测试中，输入的字符串是“pidancode.com”，因此程序正常输出“字符串正确”。在第二个测试中，输入的字符串是“皮蛋编程”，与预期的字符串不一致，因此程序抛出了一个 AssertionError 异常，并打印出了我们指定的错误信息。

使用 assert 语句可以帮助我们快速定位代码中的问题，并提供有用的错误信息，从而更容易地进行调试。