Selenium自动化测试入门：python unittest 单元测试框架

unittest又名PyUnit， Python单元测试框架（The Python unit testing framework），简称为PyUnit。自从 Python 2.1 版本后，PyUnit成为 Python标准库的一部分。

为什么需要使用unittest单元测试框架？

当我们写的用例越来越多时，我们就需要考虑用例编写的规范、组织与管理，以便于后期的维护，而python-unittest正是这样一款工具。我们这里用一个简单示例来展示用unittest脚本是什么样子的。

创建计算器类calculator

# Calculator.py
class calculator(object):
   def __init__(self, a, b):
       self.a = a
       self.b = b
 
   def add(self):
       return (self.a + self.b)
 
   def minus(self):
       return (self.a - self.b)
 
   def multip(self):
       return (self.a * self.b)
 
   def divide(self):
       return (self.a / self.b)

创建一个简单的单元测试用例

import unittest  # 导入unittest  包
from unittest_doc.com.Calculator.Calculator import calculator  # 引入需要测试的包
 
# 所有用例需继承TestCase类或者其子类
class simple_test(unittest.TestCase):
   def setUp(self):
       print('@@@初始化test_simple@@@')
       self.a = calculator(1, 2)
 
   def test_add(self):
       print('---测试用例test_simple add---')
       self.assertEqual(self.a.minus(), -1, '两值不相等')
       self.assertEqual(self.a.add(), 3, '两值不相等')
       self.assertNotEqual(self.a.divide(), 1, '两值不相等')
 
   def test_divide(self):
       print('---测试用例test_simple divide---')
       self.assertEqual(self.a.divide(), 0.5)
 
   def tearDown(self):
       print('@@@结束test_simple@@@')
 
if __name__ == '__main__':
   unittest.main()

运行结果：

@@@初始化test_simple@@@
---测试用例test_simple add---
@@@结束test_simple@@@
@@@初始化test_simple@@@
---测试用例test_simple divide---
@@@结束test_simple@@@

测试用例由setUp(),test_add(),test_divide(),tearDown() 4个部分组成，它们放在一个继承于unittest.TestCase 的测试类下。

仔细观察输出结果，每个用例运行前都会运行 setUp()，结束后会运行tearDown()，从这里应该能够看出他们两的作用。

setUp() 用于测试用例执行前的准备工作。
如测试用例中需要访问数据库，可以在setUp中建立数据库连接并进行初始化；用例需要使用web，可以先实例化浏览器；app测试需先要启动app，可先实例化app。

tearDown() 用于用例执行之后的善后工作。如关闭数据库连接，关闭浏览器，关闭app。当用例没运行成功是他也会执行。

test_add(),test_divide() 测试脚本，必须已test开头或者结尾，如用unittest.main()方法执行用例，需要已test开头。

一个TestCase 的实例就是一个测试用例。测试用例就是一个完整的测试流程，包括测试前准备环境的搭建(setUp)，实现测试过程的代码(run)，以及测试后环境的还原(tearDown)。

单元测试(unittest)的本质也就在这里，一个测试用例是一个完整的测试单元，通过运行这个测试单元，可以对某一个功能进行验证。

同时，在这我为大家准备了一份软件测试视频教程（含面试、接口、自动化、性能测试等），就在下方，需要的可以直接去观看。