分钟测试：哪个Python热门单元测试框架适合现在的你？

lsekfe

前言
　　对于我们测试来说，做自动化测试需要编写测试用例，编写测试用例就会使用到单元测试模块，常见的python单元测试模块有unittest，pytest，nose等，但是听说过最多的也就属于pytest和unittest了，安静今天就通过这一篇文字简单的介绍下pytest和unittest的区别。

　　unittest
　　说起单元测试，安静立刻想到的就是unittest框架，为什么呢？因为安静刚开始学习的时候也是学习的unittest框架。unittest属于python的内置框架，被广泛的应用到各个项目中。unittest最早的时候是受到junit的启发，unittest支持多种自动化测试用例的编写，以及支持用例前置条件和后置数据清理功能也可以将多个测试用例放在测试集中生成测试报告。

　　pytest
　　pytest也是基于python的一个单元测试框架，是基于unittest的一个扩展框架，比unittest更加简洁，方便，pytest最主要可以支持第三方插件内容，可以更加高效的完成日常工作。而且pytest也支持unittest的代码框架内容。

　　区别
　　安静简单的从几个方面进行来说明pytest和unittest之间的区别内容吧。

　　断言方面
　　unittest的断言是根据自身携带的断言内容，提供了assertEqual，assertTrue，assertFlase等断言函数。
　　pytest的断言是根据Python自带的assert断言内容进行使用的。

1. import pytest
   
2. class Test_01:
   
3.     def test_01(self):
   
4.         a = 'anjing'
   
5.         b = 'anjing'
   
6.         assert a == b
   
7.     def test_02(self):
   
8.         '''
   
9.         a是否存在b中
   
10.         '''
    
11.         a = 'anjing'
    
12.         b = 'test_anjing'
    
13.         assert a in b
    
14.     def test_03(self):
    
15.         '''
    
16.         结果是否为true
    
17.         '''
    
18.         a = True
    
19.         assert a
    
20.     def test_04(self):
    
21.         '''
    
22.         结果是否为false
    
23.         '''
    
24.         a = False
    
25.         assert a== False
    
26. if __name__ == '__main__':
    
27.     pytest.main(['-sv'])

复制代码

用例执行编写规则
　　unittest：可以使用自定义测试类内容，但必须继承unittest.TestCase的方法，测试用例需要以test的方法进行编写，用例的执行顺序是根据ASCII的顺序进行执行的。目前不能指定特定的用例顺序。unittest中提供了TestCase（测试用例），TestSuite（测试套件)，TestLoder（加载用例），TextTestRunner（执行用例）等方法让测试用例更加方便编写，编写测试用例必须导入unittest模块，执行用例必须带有unittest.main()参数。

　　pytest：测试文件名必须以test_开头的py文件或者以*_test.py结尾的py文件，测试类名必须以Test开头，测试用例必须以test_开头。Pytest可以进行执行unittest的用例，pytest的执行顺序可以通过第三方插件进行定制顺序，默认是通过从上往下的顺序进行执行。pytest的执行用例不需要导入模块。执行用例可以通过命令行的形式进行执行。

前后置操作
　　unittest：可以通过setup()和tearDown()的方法来控制用例的前后置操作，并且每条用例执行前后都会执行前后置操作。通过setupclass()和teardownclass()方法来控制一个class下的所有用例都只执行1次前后置操作。

1. import unittest
   
2. 
   
3. class Test(unittest.TestCase):
   
4.     def setUp(self):
   
5.         print('unittest前置操作，每次执行用例都会进行执行')
   
6.     def tearDown(self):
   
7.         print('unittest后置操作，每次执行用例都会进行执行')
   
8.     @classmethod
   
9.     def setUpClass(cls):
   
10.         print('所有用例只执行一次前置操作')
    
11.     @classmethod
    
12.     def tearDownClass(cls):
    
13.         print('所有用例只执行一次后置操作')
    
14.     def test_01(self):
    
15.         print('用例01')
    
16.     def test_02(self):
    
17.         print('用例02')
    
18.     def test_03(self):
    
19.         print('用例03')
    
20. 
    
21. if __name__ == '__main__':
    
22.     unittest.main()

复制代码

通过执行后发现，setupclass下的操作只执行了用例操作，setup每条用例执行前都会执行。

pytest：模块级别的前后置操作（setup_module，teardown_function）表示模块下的测试用例只执行1次前后置操作。

1. import pytest
   
2. def setup_module():
   
3.     print('每次用例执行前，只执行一次测试前置操作')
   
4. def teardown_module():
   
5.     print('每次用例执行后，只执行一次测试后置操作')
   
6. def test01():
   
7.     print('用例01')
   
8. def test02():
   
9.     print('用例02')
   
10. def test03():
    
11.     print('用例03')
    
12. if __name__ == '__main__':
    
13.     pytest.main(['-vs'])

复制代码

函数级别的前后置操作（setup_function，teardown_function）每次执行用例时都需要执行一次前后置操作。

类级别前后置操作（setup_class，teardown_class）表示类级别中测试用例只执行1次前后置操作。

方法级别前后置操作（setup_method，teardown_method）表示是在方法级别中每次都执行的用例的前后置操作。

　从上面的多种setup和teardown可以看出来很方便我们进行编写测试用例，但是pytest不仅仅是这些。pytest中有个固件fixture通过装饰器进行使用，使用范围更加广泛，可以随意命名。可以在fixture中编写前后置操作，通过yield进行区别前置操作和后置操作内容。

1. import pytest
   
2. @pytest.fixture()
   
3. def login():
   
4.     print('测试用例前置操作')
   
5.     yield
   
6.     print('测试用例后置操作')
   
7. class Test:
   
8. 
   
9.     def test01(self, login):
   
10.         print('用例01')
    
11.     def test02(self,login):
    
12.         print('用例02')
    
13.     def test03(self):
    
14.         print('用例03')
    
15. 
    
16. if __name__ == '__main__':
    
17.     pytest.main(['-vs'])

复制代码

通过执行发现，我们在用例03中没有加入fixture，所有他没有执行一些用例的前置和后置操作。

测试报告
　　unittest：unittest中没有自带的测试报告，需要下载第三方的插件HTMLTestRunner和BeautifulReport来生成详细的测试报告。
　　pytest：pytest中也没有自带的测试报告，需要下载第三方插件pytest-html或者allure-pytest进行生成详细的测试报告。

1. class Test01:
   
2.     def test_01(self):
   
3.         print('登录用例01')
   
4.     def test_02(self):
   
5.         print('编写用例02')
   
6.     def test_03(self):
   
7.         print('删除用例03')

复制代码

通过在命令中输入命令：pytest --html=report.html进行生成报告。

参数化功能
　　unittest：unittest本身不支持参数化功能，需要通过第三方库DDt来进行完成参数化功能。
　　pytest：pytest的参数化有两种方式，一种通过@pytest.mark.pyparametrize的方法，也可以通过@pytest.fixture()中的方法params来完成参数化内容。

　　失败重跑
　　unittest：目前unittest暂时不支持用例失败重跑机制
　　pytest：可以依赖第三方插件pytest-rerunfailures来进行用例失败重跑

1. import pytest
   
2. import random
   
3. class Test01:
   
4.     @pytest.mark.flaky(reruns=3, reruns_delay=2)
   
5.     def test_01(self):
   
6.         a = random.randint(0, 3)
   
7.         print('通过随机数来确保用例会有几率失败')
   
8.         print('随机数是：{}'.format(a))
   
9.         assert a == 2
   
10. 
    
11. if __name__ == '__main__':
    
12.     pytest.main(['-vs'])

复制代码

执行代码，发现断言失败后会进行重新执行。

跳过用例
　　当明知测试用例存在bug或者说该用例进行修改过了，本次执行用例时候，不想执行用例，我们可以进行跳过用例。unittest和pytest都有自带的跳过用例方法。安静通过代码实例进行介绍
　　unittest中可以通过装饰器skip的方法进行强制跳过，通过skipif的方法加上条件进行跳过，当条件达到是，进行跳过，条件未达到，不进行跳过用例
　　pytest中的跳过和unittest基本相似，跳过用例，一样要输入一些条件。

　　实战演示
　　上面简单的列举了使用的各个方法，其中有一些区别没有举例给大家说明，安静这里简单的通过请求天气和查询身份证接口的形式，编写一些测试用例，将上述没有介绍的内容，给大家通过实例的方式进行连接起来。

　　unittest
　　这里安静使用到unittest的参数化是通过ddt的形式，测试数据没有单独存放，直接放到了类外面，安静这里创建了两个接口（查询天气和查询身份证）使用了跳过用例的方法，并对测试接口进行了断言操作，并通过HTMLTestRunner的方法生成了测试报告：

1. import unittest
   
2. import requests
   
3. import HTMLTestRunner_cn
   
4. import ddt
   
5. 
   
6. # 天气接口的参数化数据
   
7. data = [{"city": '上海', "key": "331eab8f3481f37868378fcdc76cb7cd", 'result':"查询成功!"},
   
8.         {"city": "上海", "key": "331eab8f3481f37868378fcdc76cb7c", 'result':"错误的请求KEY"},
   
9.         {"city": "上", "key": "331eab8f3481f37868378fcdc76cb7cd", 'result':"暂不支持该城市"}]
   
10. 
    
11. @ddt.ddt
    
12. class Test_(unittest.TestCase):
    
13.     def tianqi(self,city,key):
    
14.         '''天气接口'''
    
15.         data = {
    
16.             "key":key,
    
17.             "city":city
    
18.         }
    
19.         r = requests.post(url='http://apis.juhe.cn/simpleWeather/query',data=data)
    
20.         return r.json()
    
21.         
    
22.     def shenfenzheng(self,cardno,key):
    
23.         '''身份证查询接口'''
    
24.         data = {
    
25.             "cardno":cardno,
    
26.             "key":key
    
27.         }
    
28.         r = requests.post('http://apis.juhe.cn/idcard/index',data= data)
    
29.         return r.json()
    
30.         
    
31.     @ddt.data(*data)
    
32.     def test_01(self,data):
    
33.         '''参数化的天气接口'''
    
34.         x = self.tianqi(city=data['city'], key=data['key'])
    
35.         self.assertEqual(x['reason'], data['result'])
    
36.         
    
37.     def test_02(self):
    
38.         '''正确的身份证号正确的key'''
    
39.         cardno = '130428197411155947'   # 身份证信息通过Faker随机创建
    
40.         key = "f40a75704fac353952a6534a18f9f437"
    
41.         # 请求查询身份证接口
    
42.         a = self.shenfenzheng(cardno,key)
    
43.         self.assertIn(a['reason'], '成功的返回')
    
44.         
    
45.     @unittest.skip('强制跳过，不需要条件')
    
46.     def test_03(self):
    
47.         '''正确的身份证号错误的key（跳过用例）'''
    
48.         cardno = '130428197411155947'
    
49.         key = "f40a75704fac353952a6534a18f9f43"
    
50.         a = self.shenfenzheng(cardno, key)
    
51.         self.assertEqual(a['reason'], '错误的请求KEY')
    
52.         
    
53.     @unittest.skipIf(True, '条件成立时候，进行跳过')
    
54.     def test_04(self):
    
55.         '''错误的身份证号正确的key（跳过用例）'''
    
56.         cardno = '42082120031108929'
    
57.         key = "f40a75704fac353952a6534a18f9f437"
    
58.         a = self.shenfenzheng(cardno, key)
    
59.         self.assertEqual(a['reason'], '请输入正确的15或18位身份证')
    
60. 
    
61. if __name__ == '__main__':
    
62.     report_path = 'report.html'
    
63.     # 打开报告
    
64.     fp = open(report_path, 'wb')
    
65.     # 报告详情
    
66.     runner = HTMLTestRunner_cn.HTMLTestRunner(stream=fp,
    
67.                                               title=u'自动化测试报告,测试结果如下：',
    
68.                                               description=u'用例执行情况：')
    
69.     # 实例化
    
70.     testunit = unittest.TestSuite()
    
71.     # 加载用例
    
72.     testunit.addTests(unittest.TestLoader().loadTestsFromTestCase(Test_))
    
73.     # 执行用例
    
74.     runner.run(testunit)
    
75.     # 关闭报告
    
76.     fp.close()

复制代码

　通过执行代码后发现已经可以成功的生成测试报告。

pytest
　　pytest这里安静也是通过上述的内容进行了修改，只是这里使用的框架是pytest。但是实现内容是一样的。天气接口进行参数化，身份证接口中2个跳过。在生成测试报告，这里安静生成的测试报告是通过allure来生成的：

1. import requests
   
2. import json
   
3. import pytest
   
4. # 参数不同值
   
5. # 天气接口的参数化数据
   
6. data = [({"city": '上海', "key": "331eab8f3481f37868378fcdc76cb7cd", 'result':"查询成功!"}),
   
7.         ({"city": "上海", "key": "331eab8f3481f37868378fcdc76cb7c", 'result':"错误的请求KEY"}),
   
8.         ({"city": "上", "key": "331eab8f3481f37868378fcdc76cb7cd", 'result':"暂不支持该城市"})]
   
9. class TestCase:
   
10.     def weather(self, city, key):
    
11.         url = 'http://apis.juhe.cn/simpleWeather/query'
    
12.         # 查询天气接口参数
    
13.         data = {
    
14.             'city': city,
    
15.             'key': key
    
16.         }
    
17.         r = requests.post(url, data=data)
    
18.         return r.json()
    
19.     def shenfenzheng(self,cardno,key):
    
20.         '''身份证查询接口'''
    
21.         data = {
    
22.             "cardno":cardno,
    
23.             "key":key
    
24.         }
    
25.         r = requests.post('http://apis.juhe.cn/idcard/index',data= data)
    
26.         return r.json()
    
27.     @pytest.mark.parametrize('data',data)
    
28.     def test_01(self,data):
    
29.         # 调用天气预报接口
    
30.         r = TestCase().weather(city= data['city'],key = data['key'])
    
31.         assert r['reason'] == data['result']
    
32.     def test_02(self):
    
33.         '''正确的身份证号正确的key'''
    
34.         cardno = '130428197411155947'   # 身份证信息通过Faker随机创建
    
35.         key = "f40a75704fac353952a6534a18f9f437"
    
36.         # 请求查询身份证接口
    
37.         a = self.shenfenzheng(cardno,key)
    
38.         assert a['reason'] == '成功的返回'
    
39.     @pytest.mark.skip('强制跳过，不需要条件')
    
40.     def test_03(self):
    
41.         '''正确的身份证号错误的key（跳过用例）'''
    
42.         cardno = '130428197411155947'
    
43.         key = "f40a75704fac353952a6534a18f9f43"
    
44.         a = self.shenfenzheng(cardno, key)
    
45.         assert  a['reason'] =='错误的请求KEY'
    
46.     @pytest.mark.skipif(True, reason='条件成立时候，进行跳过')
    
47.     def test_04(self):
    
48.         '''错误的身份证号正确的key（跳过用例）'''
    
49.         cardno = '42082120031108929'
    
50.         key = "f40a75704fac353952a6534a18f9f437"
    
51.         a = self.shenfenzheng(cardno, key)
    
52.         assert a['reason'] in '请输入正确的15或18位身份证'
    
53. if __name__ == '__main__':
    
54.     pytest.main(['-s'])

复制代码

这里我们通过在cmd中先执行pytest --alluredir report进行打开报告，然后在通过allure serve report打开测试报告。从报告中就能看到已经全部都执行成功了，且用例跳过了2条。

总结
　　安静简单的列举出来了几点并通过项目实例进行对比两种框架的情况。其中我们可以看出来unittest编写用例比pytest复杂，最最最主要的一点是unittest没有较多的第三方插件支持，但是pytest可以支持更多的第三方插件，这样可以更加方便的进行投入到编写测试用例的过程中。当然对于新手来说，安静还是推荐先使用unittest，看看unittest的源码内容，当unittest学会之后在去接触pytest，这样就更加容易上手。其次unittest作为Python中的内置框架，也是值得大家进行学习的。