Python Unit Test 单元测试

软件测试中分很多种，常见如下：

• 压力测试 Stress Test
• 系统测试 System Test
• 集成测试 Integration Test
• 单元测试 Unit Test

其中Unit Test单元测试是专门给软件开发时用的辅助减少bug的测试，也是我们最常用的。

单元测试中的单元一般都是极小极小的单元，小到不可再分割那种。比如测试一个计算平方的函数是否能够正确返回平方值。

断言 Assertion

Python关键字assert用来判断一个表达式是否为True。如果True就没问题，如果False则立马抛出一个AssertionException的异常。

如：

assert 1 == 1

assert 45 in [12, 34, 45]

Test Case 测试用例

我们可以自己用assert关键字去手写全部测试逻辑，我们也可以直接用Python中内置的一个专门用来测试的类TestCase测试用例来完成大量的准备工作，我们只需要写测试函数即可。

TestCase中封装好了很多测试的流程和工具，而我们只需要自己定义一个类来继承它，然后简单写上每个测试函数即可自动完成所有工作。

TestCase中定义了一些hooks，即测试前和测试后分别执行的函数，它还定义了一些方便的assertion语句，如assertIn，assertEqual等。

一个标准的测试模块定义如下：

import unittest

class MyTest(unittest.TestCase):
    
    # 重写父类方法：在测试开始前会执行此函数
    def setUp(self):
        pass

    # 重写父类的方法：在测试完成后会执行此函数
    def tearDown(self):
        pass

    # 具体的测试函数：规定以"test_"开头才会被执行
    def test_is_correct_number(self):
        # 具体的测试代码
        pass

注意：测试模块一般保存为test_*.py。

怎么执行测试呢？只要在命令行里直接执行即可：

$ python test_abc.py

测试函数编写

具体怎么写一个测试函数呢？

测试一个某个变量是否在某个列表中：

import unittest

from someModule import myVariable, myList

class MyTest(unittest.TestCase):
    def test_is_variable_in_list(self):
        self.assertIn(myVariable, myList)

以上代码，如果myVariable in myList返回的true，则测试成功，否则失败，且assertion失败后的语句都不会被执行。

测试一个函数是否能正常返回正确的值：

#...
def calculate_square(num):
    return num**2

class MyTest(unittest.TestCase):

    def test_calculate_square(self):
        result = calculate_square(4)
        self.assertEqual(result, 16)