使用unittest编写Python单元测试需创建继承自TestCase的类，测试方法以test_开头，通过断言方法验证逻辑。例如为calculator模块编写TestCalculator类，用assertEqual、assertRaises等方法测试加减乘除函数，确保正常与异常情况均被覆盖。命令行运行python test_calculator.py即可执行测试，-v参数显示详细结果。
写Python单元测试主要用内置的unittest框架，也可以选择第三方库如pytest。这里以unittest为例，介绍编写方法和实际例子，帮助你快速上手。
1. 理解单元测试的基本结构
unittest是Python标准库中的测试框架，基于类和方法组织测试。每个测试用例继承自unittest.TestCase，测试方法名必须以test_开头。
基本结构如下：
import unittestclass TestMathFunctions(unittest.TestCase):def test_addition(self):self.assertEqual(2 + 2, 4)
if name == 'main':unittest.main()
2. 编写被测函数和对应测试
假设我们有一个简单的计算器模块calculator.py：
立即学习“Python免费学习笔记（深入）”；
# calculator.pydef add(a, b): return a + bdef subtract(a, b):return a - b
def multiply(a, b):return a * b
def divide(a, b):if b == 0:raise ValueError("Cannot divide by zero.")return a / b
接下来为它写测试文件test_calculator.py：
# test_calculator.pyimport unittestfrom calculator import add, subtract, multiply, divideclass TestCalculator(unittest.TestCase):
def test_add(self): self.assertEqual(add(2, 3), 5) self.assertEqual(add(-1, 1), 0) self.assertEqual(add(0, 0), 0)def test_subtract(self): self.assertEqual(subtract(5, 3), 2) self.assertEqual(subtract(0, 5), -5)def test_multiply(self): self.assertEqual(multiply(3, 4), 12) self.assertEqual(multiply(-2, 3), -6)def test_divide(self): self.assertEqual(divide(10, 2), 5) self.assertAlmostEqual(divide(7, 3), 2.333333, places=6)def test_divide_by_zero(self): with self.assertRaises(ValueError): divide(10, 0)
if name == 'main':unittest.main()
3. 常用断言方法说明
TestCase类提供多种断言方法验证结果：
assertEqual(a, b)：检查a是否等于bassertNotEqual(a, b)：检查a不等于bassertTrue(x)：检查x为真assertFalse(x)：检查x为假assertIn(a, b)：检查a在b中assertIsInstance(obj, cls)：检查对象是否为某类实例assertRaises(exception)：检查是否抛出指定异常assertAlmostEqual(a, b)：检查浮点数近似相等
4. 运行测试
在命令行执行测试文件：
python test_calculator.py
输出会显示运行了多少个测试，是否全部通过。如果某个测试失败，会指出具体哪一行出错。
加上参数-v可查看详细结果：
python test_calculator.py -v
基本上就这些。掌握结构、写好断言、覆盖正常和异常情况，就能写出有效的单元测试。
以上就是Python单元测试怎么写_Python单元测试编写方法与实例的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1379180.html
微信扫一扫 
支付宝扫一扫 
