25 python入门指南如何编写测试代码

Posted awildfish

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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("
TestMathFunc: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

以上是关于25 python入门指南如何编写测试代码的主要内容,如果未能解决你的问题,请参考以下文章

使用 Python 代码片段编写 LaTeX 文档

Ansible项目大牛:给Python软件开发测试的25个忠告!

在 Python 多处理进程中运行较慢的 OpenCV 代码片段

测试新人如何使用Python代码封装自动化测试的用例?

Python代码阅读(第25篇):将多行字符串拆分成列表

Flask 编写http接口api及接口自动化测试