你会写单元测试吗

关于我
编程界的一名小小程序猿，目前在一个创业团队任team lead，技术栈涉及Android、Python、Java和Go，这个也是咱们团队的主要技术栈。 联系：hylinux1024@gmail.com

也许你已经据说过Test Driven Development，但不知道你是否遵循这个规则呢？其实我本身在写代码的时候也不多会先写单元测试再写业务功能逻辑。这不我也今天也来学习如何在Python中写单元测试。

0x00 unittest

Python中的unittest单元测试框架跟其它语言如JUnit是相似的。它支持测试自动化、配置共享以及关机代码测试。

假设在个人项目目录下有一个mysum模块用于计算列表中各个数之和。 还有一个test_mysum.py用于编写单元测试的文件。

myproject/
│
├── mysum/
│   └── __init__.py
└── unittests
    └── test_mysum.py

复制代码

打开mysum模块中的__init__.py文件。

添加下面的方法

def sum(args):
    total = 0
    for arg in args:
        total += arg
    return total
复制代码

打开test_mysum.py，编写单元测试

import unittest

from mysum import sum

class TestSum(unittest.TestCase):
    def test_list_int(self):
        """ 测试一个整数列表的和 """
        data = [1, 2, 3]
        result = sum(data)
        self.assertEqual(result, 6)


if __name__ == '__main__':
    unittest.main()

复制代码

首先导入咱们要测试的模块mysum 测试用例类经过继承unittest.TestCase来实现，测试方法test_list_int是以test开头的。 在这个方法中定义了一个整型列表，执行sum方法，而后判断执行结果是否与预期相符。

最后调用unittest.main()来执行这个测试用例

Ran 1 test in 0.001s

OK
复制代码

若是再添加一个方法

def test_list_sum(self):
    data = [1, 3, 4]

    result = sum(data)
    self.assertEqual(result, 6)
复制代码

执行后会看到以下相似信息

6 != 8

Expected :8
Actual   :6
复制代码

从这个输出信息能够看出指望值与实际值不相符，这时候若是咱们的测试用例没有问题，那就要看看mysum的实现逻辑了。

从这个例子中能够总结一个测试用例的过程：

1. 构建输入数据
2. 执行要测试模块，获取执行结果
3. 与预期结果相比较，根据结果修改代码

0x01 setup/tearDown

在编写单元测试时，还能够重写父类的setup和tearDown方法，能够在执行测试逻辑开始前和结束时作一些处理。例如在setup方法中能够初始化测试数据，在tearDown方法作一些清理工做。

import unittest

class TestBasic(unittest.TestCase):
    def setUp(self):
        # 加载测试数据
        self.app = App(database='fixtures/test_basic.json')

    def test_customer_count(self):
        self.assertEqual(len(self.app.customers), 100)

    def test_existence_of_customer(self):
        customer = self.app.get_customer(id=10)
        self.assertEqual(customer.name, "Org XYZ")
        self.assertEqual(customer.address, "10 Red Road, Reading")
    
    def tearDown(self):
        self.app.releaseDB()
复制代码

在每一个测试执行时setup和tearDown都会被执行一次。

0x02 pytest

pytest是一个第三方测试框架，使用它不须要继承某个类，它可使用原生的assert语句用于测试结果的断言。

它的用法也很简单

首先经过pip安装

➜ pip install pytest
复制代码

咱们写一个单独的tests文件夹下建立测试用例文件test_pytest.py。 注意：这里pytest的文件必须与上文的unittests文件必须区分开，不然会出现ModuleNotFoundError。我已经在这里踩坑。

我这里使用pytest单元测试的文件结构为

tests
│
└── test_func.py
复制代码

test_func.py的内容为

# 导入咱们要测试的模块
from mysum import sum


def test_answer():
    data = [1, 2, 3]

    assert sum(data) == 5

复制代码

测试方法以test开头

而后再命令行中执行

➜ python -m pytest tests/test_func.py 
复制代码

注意：这里要使用python -m pytest，若是直接使用pytest会提示ModuleNotFoundError: No module named 'mysum'

执行结果以下

tests/test_func.py F                                                                                                                                                                          [100%]

======================================================= FAILURES ==============================================
_______________________________________________________ test_answer ___________________________________________

    def test_answer():
        data = [1, 2, 3]
    
>       assert sum(data) == 5
E       assert 6 == 5
E        +  where 6 = sum([1, 2, 3])

tests/test_func.py:11: AssertionError
=======================================================1 failed in 0.03 seconds =======================================================
复制代码

因为6!=5，这个单元测试提示了出错的位置。

0x03 总结一下

测试代码对编写代码很是重要，写单元测试也是一个好习惯。本文也只是一个开胃菜。要写出健壮的代码，从写单元测试开始吧。

0x04 学习资料

• docs.python.org/3/library/u…
  Unit testing framework
• realpython.com/python-test…
  Getting Started With Testing in Python
• docs.python-guide.org/writing/tes… Testing Your Code