25 【python入门指南】如何编写测试代码
python如何编写测试代码
python内置了unittest，使得写应用层的单元测试变得超乎寻常的简单。
1，执行单个测试函数
#!/bin/python import unittest class TestMathFunc(unittest.TestCase): def test_add(self): self.assertEqual(3, 1+2) self.assertEqual(4, 2+2) self.assertNotEqual(3, 1+3) def runTest(self): self.test_add() suite = unittest.TestSuite() testCase = [TestMathFunc()] suite.addTests(testCase) runner = unittest.TextTestRunner(verbosity = 2) runner.run(suite)
单元测试，只需要仿照上面的例子写就可以完成自己的测试代码。
其中包含几个部分：
a，测试类（要继承unittest.TestCase）；
b，测试小组件，unittest.TestSuite，负责将测试类装入进来；
c，执行类，unittest.TextTestRunner；
2，执行多个测试函数
#!/bin/python import unittest class TestMathFunc(unittest.TestCase): def test_add(self): self.assertEqual(3, 1+2) self.assertEqual(4, 2+2) self.assertNotEqual(3, 1+3) def test_minus(self): self.assertEqual(-1, 1-2) def runTest(self): self.test_add() self.test_minus() suite = unittest.TestSuite() #testCase = [TestMathFunc('test_add'), TestMathFunc('test_minus')] testCase = [TestMathFunc()] suite.addTests(testCase) runner = unittest.TextTestRunner(verbosity = 2) runner.run(suite)
修改runTest函数，将测试函数包含进来即可。
3，执行多个测试类
#!/bin/python import unittest class TestStringFunc(unittest.TestCase): def test_str_concat(self): self.assertEqual("abcabc", "abc" + "abc") def runTest(self): self.test_str_concat() class TestMathFunc(unittest.TestCase): def test_add(self): self.assertEqual(3, 1+2) self.assertEqual(4, 2+2) self.assertNotEqual(3, 1+3) def test_minus(self): self.assertEqual(-1, 1-2) def runTest(self): self.test_add() self.test_minus() suite = unittest.TestSuite() testCase = [TestMathFunc(), TestStringFunc()] suite.addTests(testCase) runner = unittest.TextTestRunner(verbosity = 2) runner.run(suite)
在suite添加多个测试类，即可。
当然还有另外的方式：
... suite = unittest.TestSuite() testCase = [TestMathFunc()] suite.addTests(testCase) testCase2 = TestStringFunc() suite.addTest(testCase2) runner = unittest.TextTestRunner(verbosity = 2) runner.run(suite)
通过以上代码片段，即可实现，自主地控制多个测试类，自主地控制多个函数执行测试，十分方便。
4，有时候我们不想每个测试函数都显示调用，而是加载所有test_开头的测试函数，怎么做？
构建两个文件：t3.py，main.py
main文件存放执行代码，t3文件仅存放测试类，不存放任何执行代码
main.py
#!/bin/python
#main.py
import unittest from t3 import TestMathFunc, TestStringFunc suite2 = unittest.TestLoader().discover('.', 't3.py') runner = unittest.TextTestRunner(verbosity = 2) runner.run(suite2)
t3.py
#!/bin/python #t3.py import unittest class TestStringFunc(unittest.TestCase): def test_str_concat(self): self.assertEqual("abcabc", "abc" + "abc") # def runTest(self): # self.test_str_concat() class TestMathFunc(unittest.TestCase): def test_add(self): self.assertEqual(3, 1+2) self.assertEqual(4, 2+2) self.assertNotEqual(3, 1+3) def test_minus(self): self.assertEqual(-1, 1-2) # def runTest(self): # self.test_add() # self.test_minus()
执行main.py，输出结果：
test_add (t3.TestMathFunc) ... ok test_minus (t3.TestMathFunc) ... ok test_str_concat (t3.TestStringFunc) ... ok ---------------------------------------------------------------------- Ran 3 tests in 0.001s
通过 unittest.TestLoader().discover('.', 't3.py') 将t3文件中包含的所有测试类导入到suite中。
依赖这个功能，我们可以将所有的需要测试的文件，导入到main主函数中，进行集中管理和测试。注意：这里不再依赖runTest函数，只要函数名中包含test前缀就会被包含进来。
5，有时候测试用例需要提供上下文环境，怎么做？
我们针对例子4，仅修改t3.py测试类。
class TestMathFunc(unittest.TestCase): def setUp(self): print("\nTestMathFunc:setup") self.a = 1 self.b = 2 def test_add(self): self.assertEqual(3, 1+2) self.assertEqual(4, 2+2) self.assertNotEqual(3, 1+3) self.assertEqual(3, self.a + self.b) def test_minus(self): self.assertEqual(-1, 1-2)
这里我们准备了环境信息（准备两个变量a和b），使用这两个变量进行单元测试。
test_add (t3.TestMathFunc) ... TestMathFunc:setup ok test_minus (t3.TestMathFunc) ... TestMathFunc:setup ok test_str_concat (t3.TestStringFunc) ... ok ---------------------------------------------------------------------- Ran 3 tests in 0.002s OK
函数的输出如上。
类的单元测试，每个函数执行之前都会执行setUp函数。
对应的是tearDown，用来销毁构建的环境信息。可以自己尝试下。
参考网站：
更详细的解释：https://www.jianshu.com/p/38948d0d73f5
最详细的官方文档：https://docs.python.org/3/library/unittest.html#module-unittest
