前言
单元测试是验证代码正确性的"质检工具"。
Python自带的unittest
模块,其中assertEqual
、assertTrue
、assertFalse
这三个基础断言方法,是入门单元测试的"敲门砖"。今天就给大家分享一下它们的用法。
一、unittest
与断言,是代码的"质检组合"
首先要明确:unittest
是Python内置的单元测试框架,能帮我们自动化验证代码逻辑是否正确;而"断言方法"就是这个框架里的"判断标准"------就像质检员检查产品时,会用"尺寸是否达标""材质是否合格"作为标准,断言方法则用"值是否相等""条件是否为真"来判断代码输出是否符合预期。
简单说,当你写了一个函数(比如计算两数之和),用断言方法就能自动检查"输入3+2,输出是不是5",不用手动运行代码、盯着结果看。如果断言通过,说明代码逻辑没问题;如果不通过,会明确提示"哪里错了",帮你快速定位bug。
二、assertEqual(a, b)
------检查"两个东西是不是一模一样"
assertEqual
的作用很直接:判断a
和b
是否相等(即a == b
),如果相等,测试通过;如果不相等,测试失败并提示差异。它就像检查"两个苹果是不是一样重""两张身份证号是不是完全相同",是最常用的断言方法之一。
代码示例
python
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
),如果是,测试通过;如果不是,测试失败。比如检查"今天是不是晴天""这个学生的分数是不是及格了",适合验证"某个条件是否成立"。
代码示例
python
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
),如果是,测试通过;如果不是,测试失败。比如检查"今天是不是下雨""这个产品是不是过期了",适合验证"某个条件一定不成立"。
代码示例(可直接复制运行)
python
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()
,这样直接运行脚本就能执行所有测试用例。
六、基础断言,是单元测试的"第一步"
assertEqual
、assertTrue
、assertFalse
这三个方法,是Python单元测试的"基础工具",能够帮你自动化验证代码逻辑,避免"手动测试漏看bug""改了代码后旧问题复发"。