如果可能的话,代码库中的所有代码都要测试。但这取决于开发者,如果写一个健壮性测试是不切实际的,你可以跳过它。就像_NickCoghlan_(Python核心开发成员)在访谈里面说的:有一个坚实可靠的测试套件,你可以做出大的改动,并确信外部可见行为保持不变。
这里引用维基百科的介绍:
在计算机编程中,单元测试(英语:UnitTesting)又称为模块测试,是针对程序模块(软件设计的最小单位)来进行正确性检验的测试工作。程序单元是应用的最小可测试部件。在过程化编程中,一个单元就是单个程序、函数、过程等;对于面向对象编程,最小单元就是方法,包括基类(超类)、抽象类、或者派生类(子类)中的方法。
单元测试模块
在Python里我们有unittest这个模块来帮助我们进行单元测试。
阶乘计算程序
在这个例子中我们将写一个计算阶乘的程序factorial.py:
importsysdeffact(n):"""阶乘函数:argn:数字:returns:n的阶乘"""ifn==0:return1returnn*fact(n-1)defdiv(n):"""只是做除法"""res=10/nreturnresdefmain(n):res=fact(n)print(res)if__name__=='__main__':iflen(sys.argv)>1:main(int(sys.argv[1]))运行程序:
$python3factorial.py5
测试哪个函数?
正如你所看到的,fact(n)这个函数执行所有的计算,所以我们至少应该测试这个函数。
编辑factorial_test.py文件,代码如下:
importunittestfromfactorialimportfactclassTestFactorial(unittest.TestCase):"""我们的基本测试类"""deftest_fact(self):"""实际测试任何以`test_`开头的方法都被视作测试用例"""res=fact(5)self.assertEqual(res,120)if__name__=='__main__':unittest.main()运行测试:
$python3factorial_test.py.----------------------------------------------------------------------Ran1testin0.000sOK说明
我们首先导入了unittest模块,然后测试我们需要测试的函数。
测试用例是通过子类化unittest.TestCase创建的。
现在我们打开测试文件并且把120更改为121,然后看看会发生什么?
各类assert语句
如果我们在factorial.py中调用div(0),我们能看到异常被抛出。
我们也能测试这些异常,就像这样:
self.assertRaises(ZeroDivisionError,div,0)完整代码:
importunittestfromfactorialimportfact,divclassTestFactorial(unittest.TestCase):"""我们的基本测试类"""deftest_fact(self):"""实际测试任何以`test_`开头的方法都被视作测试用例"""res=fact(5)self.assertEqual(res,120)deftest_error(self):"""测试由运行时错误引发的异常"""self.assertRaises(ZeroDivisionError,div,0)if__name__=='__main__':unittest.main()
mounttab.py中只有一个mount_details()函数,函数分析并打印挂载详细信息。
importosdefmount_details():"""打印挂载详细信息"""ifos.path.exists('/proc/mounts'):fd=open('/proc/mounts')forlineinfd:line=line.strip()words=line.split()print('{}on{}type{}'.format(words[0],words[1],words[2]),end='')iflen(words)>5:print('({})'.format(''.join(words[3:-2])))else:print()fd.close()if__name__=='__main__':mount_details()重构mounttab.py
现在我们在mounttab2.py中重构了上面的代码并且有一个我们能容易的测试的新函数parse_mounts()。
importosdefparse_mounts():"""分析/proc/mounts并返回元祖的列表"""result=[]ifos.path.exists('/proc/mounts'):fd=open('/proc/mounts')forlineinfd:line=line.strip()words=line.split()iflen(words)>5:res=(words[0],words[1],words[2],'({})'.format(''.join(words[3:-2])))else:res=(words[0],words[1],words[2])result.append(res)fd.close()returnresultdefmount_details():"""打印挂载详细信息"""result=parse_mounts()forlineinresult:iflen(line)==4:print('{}on{}type{}{}'.format(*line))else:print('{}on{}type{}'.format(*line))if__name__=='__main__':mount_details()同样我们测试代码,编写mounttest.py文件:
#!/usr/bin/envpythonimportunittestfrommounttab2importparse_mountsclassTestMount(unittest.TestCase):"""我们的基本测试类"""deftest_parsemount(self):"""实际测试任何以`test_`开头的方法都被视作测试用例"""result=parse_mounts()self.assertIsInstance(result,list)self.assertIsInstance(result[0],tuple)deftest_rootext4(self):"""测试找出根文件系统"""result=parse_mounts()forlineinresult:ifline[1]=='/'andline[2]!='rootfs':self.assertEqual(line[2],'ext4')if__name__=='__main__':unittest.main()运行程序
$python3mounttest.py..----------------------------------------------------------------------Ran2testsin0.001sOK
测试覆盖率是找到代码库未经测试的部分的简单方法。它并不会告诉你的测试好不好。
在Python中我们已经有了一个不错的覆盖率工具来帮助我们。你可以在实验楼环境中安装它:
$sudopip3installcoverage覆盖率示例
$coverage3runmounttest.py..----------------------------------------------------------------------Ran2testsin0.013sOK$coverage3report-mNameStmtsMissCoverMissing--------------------------------------------mounttab2.py22768%16,25-30,34mounttest.py140100%--------------------------------------------TOTAL36781%
我们还可以使用下面的命令以HTML文件的形式输出覆盖率结果,然后在浏览器中查看它。
$coverage3html
知识点回顾:
本节了解了什么是单元测试,unittest模块怎么用,测试用例怎么写。以及最后我们使用第三方模块coverage进行了覆盖率测试。
在实际生产环境中,测试环节是非常重要的的一环,即便志不在测试工程师,但以后的趋势就是DevOps,所以掌握良好的测试技能也是很有用的。