你会写单元测试吗

最新推荐文章于 2024-03-20 19:17:19 发布

阿星君

最新推荐文章于 2024-03-20 19:17:19 发布

阅读量127

点赞数

分类专栏： java高阶框架大厂专题面试专题面试文章标签：单元测试 java 编程语言 python unittest

本文链接：https://blog.csdn.net/dingyu002/article/details/117924301

版权

java高阶框架同时被 3 个专栏收录

72 篇文章 2 订阅

订阅专栏

专题面试

25 篇文章 0 订阅

订阅专栏

大厂专题面试

13 篇文章 0 订阅

订阅专栏

也许你已经听说过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的实现逻辑了。

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

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

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 总结一下

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

最后小编在学习过程中整理了一些学习资料，可以分享给做java的工程师朋友们，相互交流学习，需要的可以加入我的学习交流群 716055499 即可免费获取Java架构学习资料（里面有高可用、高并发、高性能及分布式、Jvm性能调优、Spring源码，MyBatis，Netty,Redis,Kafka,Mysql,Zookeeper,Tomcat,Docker,Dubbo,Nginx等多个知识点的架构资料）

作者：dingyu002

来源：dinyu002

著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。