Python单元测试入门：3个核心断言方法，帮你快速定位代码bug

匿名上传

发布时间:2025-09-11 10:30:02

本文导读

前言

Python自带的unittest模块，其中assertEqual、assertTrue、assertFalse这三个基础断言方法，是入门单元测试的“敲门砖”。今天就给大家分享一下它们的用法。

一、`unittest`与断言，是代码的“质检组合”

首先要明确：unittest是Python内置的单元测试框架，能帮我们自动化验证代码逻辑是否正确；而“断言方法”就是这个框架里的“判断标准”——就像质检员检查产品时，会用“尺寸是否达标”“材质是否合格”作为标准，断言方法则用“值是否相等”“条件是否为真”来判断代码输出是否符合预期。

简单说，当你写了一个函数（比如计算两数之和），用断言方法就能自动检查“输入3+2，输出是不是5”，不用手动运行代码、盯着结果看。如果断言通过，说明代码逻辑没问题；如果不通过，会明确提示“哪里错了”，帮你快速定位bug。

二、`assertEqual(a, b)`——检查“两个东西是不是一模一样”

assertEqual的作用很直接：判断a和b是否相等（即a == b），如果相等，测试通过；如果不相等，测试失败并提示差异。它就像检查“两个苹果是不是一样重”“两张身份证号是不是完全相同”，是最常用的断言方法之一。

代码示例

import unittest

class TestExample(unittest.TestCase):
    def test_equal(self):
        self.assertEqual(3 + 2, 5)  # 通过：3+2的结果确实是5
        self.assertEqual("hello", "hello")  # 通过：两个字符串完全相同
        self.assertEqual([1, 2], [1, 2])  # 通过：两个列表的元素和顺序都一致

解读

第一个断言：验证“3加2的结果”和“5”是否相等，这是最基础的数值判断，比如你写了个加法函数，就能用它测结果对不对；
第二个断言：验证两个“hello”字符串是否一样，适合测试字符串处理逻辑（比如“截取字符串后是不是预期结果”）；
第三个断言：验证两个列表是否相等，适合测试列表相关操作（比如“排序后的列表是不是预期顺序”）。

如果把代码里的5改成6，运行测试时就会报错，提示“3+2 != 6”，帮你立刻发现“计算逻辑错了”。

三、`assertTrue(x)`——检查“这个条件是不是真的”

assertTrue的作用是判断x是否为“真”（即x的布尔值是True），如果是，测试通过；如果不是，测试失败。比如检查“今天是不是晴天”“这个学生的分数是不是及格了”，适合验证“某个条件是否成立”。

代码示例

import unittest

class TestExample(unittest.TestCase):
    def test_boolean(self):
        self.assertTrue(1 == 1)  # 通过：1等于1，条件成立
        self.assertTrue("hello" in "hello world")  # 通过："hello"确实在"hello world"里
        self.assertTrue(100 > 50)  # 通过：100大于50，条件成立

解读

第一个断言：验证“1等于1”这个条件是否为真，看似简单，实则适合测试“变量是否等于预期值”（比assertEqual更侧重“条件判断”）；
第二个断言：验证“hello”是否包含在“hello world”里，适合测试字符串包含、列表元素存在等场景（比如“用户输入的关键词是不是在结果里”）；
第三个断言：验证“100大于50”是否为真，适合测试大小比较、范围判断（比如“计算出的数值是不是超过阈值”）。

比如你写了个“判断用户年龄是否成年”的函数，输入20，就能用self.assertTrue(age >= 18)来验证结果是否正确。

四、`assertFalse(x)`——检查“这个条件是不是假的”

assertFalse和assertTrue是“相反搭档”：它判断x是否为“假”（即x的布尔值是False），如果是，测试通过；如果不是，测试失败。比如检查“今天是不是下雨”“这个产品是不是过期了”，适合验证“某个条件一定不成立”。

代码示例（可直接复制运行）

import unittest

class TestExample(unittest.TestCase):
    def test_false(self):
        self.assertFalse(1 == 2)  # 通过：1等于2是假的，条件不成立
        self.assertFalse("python" in "hello world")  # 通过："python"不在"hello world"里
        self.assertFalse(10 < 5)  # 通过：10小于5是假的，条件不成立

解读

第一个断言：验证“1等于2”是假的，适合测试“两个值一定不相等”的场景（比如“错误输入是不是不会得到正确结果”）；
第二个断言：验证“python”不在“hello world”里，适合测试“某个元素一定不存在”（比如“过滤后的列表是不是没有无效值”）；
第三个断言：验证“10小于5”是假的，适合测试“某个比较一定不成立”（比如“计算出的数值是不是不会低于最小值”）。

比如你写了个“过滤敏感词”的函数，输入“badword”，就能用self.assertFalse("badword" in filtered_text)验证“敏感词是不是被成功过滤了”。

五、用对断言，让测试更“靠谱”

这三个断言方法虽然简单，但用的时候有个小原则：“选最贴合场景的”：

如果要判断“两个值是否相等”，优先用assertEqual（比assertTrue(a == b)更清晰，失败时提示更具体）；
如果要判断“某个条件是否成立”，用assertTrue或assertFalse（不用绕弯子写assertEqual(条件, True)）；
运行测试时，记得在代码最后加一句if __name__ == '__main__': unittest.main()，这样直接运行脚本就能执行所有测试用例。