Python单元测试怎么写_Python单元测试编写方法与实例

技术教程
2025-11-10 21:24:03

使用unittest编写Python单元测试需创建继承自TestCase的类，测试方法以test_开头，通过断言方法验证逻辑。例如为calculator模块编写TestCalculator类，用assertEqual、assertRaises等方法测试加减乘除函数，确保正常与异常情况均被覆盖。命令行运行python test_calculator.py即可执行测试，-v参数显示详细结果。

python单元测试怎么写_python单元测试编写方法与实例

写Python单元测试主要用内置的unittest框架，也可以选择第三方库如pytest。这里以unittest为例，介绍编写方法和实际例子，帮助你快速上手。

1. 理解单元测试的基本结构

unittest是Python标准库中的测试框架，基于类和方法组织测试。每个测试用例继承自unittest.TestCase，测试方法名必须以test_开头。

基本结构如下：

import unittest
<p>class TestMathFunctions(unittest.TestCase):
def test_addition(self):
self.assertEqual(2 + 2, 4)</p><p>if <strong>name</strong> == '<strong>main</strong>':
unittest.main()</p>

2. 编写被测函数和对应测试

假设我们有一个简单的计算器模块calculator.py：

# calculator.py
def add(a, b):
    return a + b
<p>def subtract(a, b):
return a - b</p><p>def multiply(a, b):
return a * b</p><p>def divide(a, b):
if b == 0:
raise ValueError("Cannot divide by zero.")
return a / b</p>

接下来为它写测试文件test_calculator.py：

# test_calculator.py
import unittest
from calculator import add, subtract, multiply, divide
<p>class TestCalculator(unittest.TestCase):</p>
                    <div class="aritcle_card">
                        <a class="aritcle_card_img" href="/ai/1866">
                            <img src="https://img.php.cn/upload/ai_manual/000/000/000/175680392413552.jpg" alt="Designify">
                        </a>
                        <div class="aritcle_card_info">
                            <a href="/ai/1866">Designify</a>
                            <p>拖入图片便可自动去除背景✨</p>
                            <div class="">
                                <img src="/static/images/card_xiazai.png" alt="Designify">
                                <span>79</span>
                            </div>
                        </div>
                        <a href="/ai/1866" class="aritcle_card_btn">
                            <span>查看详情</span>
                            <img src="/static/images/cardxiayige-3.png" alt="Designify">
                        </a>
                    </div>
                <pre class='brush:python;toolbar:false;'>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)