unittest简单上手

最新推荐文章于 2023-02-05 18:10:46 发布

杯子很干净

最新推荐文章于 2023-02-05 18:10:46 发布

阅读量234

点赞数

文章标签： unittest python

原文链接：https://www.cnblogs.com/daxiong2014/p/10449184.html

版权

一、什么是unittest

unittest是Python单元测试框架，类似于JUnit框架。

unittest中有4个重要的概念：test fixture, test case, test suite, test runner

Testcase：

一个TestCase的实例就是一个测试用例。什么是测试用例呢？就是一个完整的测试流程，包括测试前准备环境的搭建(setUp)，执行测试代码 (run)，以及测试后环境的还原(tearDown)。元测试(unit test)的本质也就在这里，一个测试用例是一个完整的测试单元，通过运行这个测试单元，可以对某一个问题进行验证。

Test suite：

多个测试用例集合在一起，就是TestSuite，而且TestSuite也可以嵌套TestSuite。

Test runner：

是来执行测试用例的，其中的run(test)会执行TestSuite/TestCase中的run(result)方法。

TestLoader：

是用来加载TestCase到TestSuite中的，其中有几个loadTestsFrom__()方法，就是从各个地方寻找TestCase，创建它们的实例，然后add到TestSuite中，再返回一个TestSuite实例。

Test fixture：

对一个测试用例环境的搭建和销毁，是一个fixture，通过覆盖 TestCase的setUp()和tearDown()方法来实现。这个有什么用呢？比如说在这个测试用例中需要访问数据库，那么可以在setUp() 中建立数据库连接以及进行一些初始化，在tearDown()中清除在数据库中产生的数据，然后关闭连接。注意tearDown的过程很重要，要为以后的 TestCase留下一个干净的环境。关于fixture，还有一个专门的库函数叫做fixtures，功能更加强大。

使用unittest编写python的单元测试代码，包括如下几个步骤：

1、编写一个python类，继承 unittest模块中的TestCase类，这就是一个测试类

2、在上面编写的测试类中定义测试方法（这个就是指的测试用例），每个方法的方法名要求以 test 打头，没有额外的参数。在该测试方法中调用被测试代码，校验测试结果，TestCase类中提供了很多标准的校验方法，如最常见的assertEqual。

3、执行 unittest.main() ，该函数会负责运行测试，它会实例化所有TestCase的子类，并运行其中所有以test打头的方法。

二、简单用法

unittest是python自带的一个单元测试框架，类似于java的junit，基本结构是类似的。基本用法如下：

1.用import unittest导入unittest模块

2.定义一个继承自unittest.TestCase的测试用例类，如class xxx(unittest.TestCase):

3.定义setUp和tearDown，这两个方法与junit相同，即如果定义了则会在每个测试case执行前先执行setUp方法，执行完毕后执行tearDown方法。

4.定义测试用例，名字以test开头，unittest会自动将test开头的方法放入测试用例集中。

5.一个测试用例应该只测试一个方面，测试目的和测试内容应很明确。主要是调用assertEqual、assertRaises等断言方法判断程序执行结果和预期值是否相符。

6.调用unittest.main()启动测试

7.如果测试未通过，则会显示e，并给出具体的错误（此处为程序问题导致）。如果测试失败则显示为f，测试通过为.，如有多个testcase，则结果依次显示。

三、unittest模块的常用方法

assertEqual(a, b) a == b

assertNotEqual(a, b) a != b

assertTrue(x) bool(x) is True

assertFalse(x) bool(x) is False

assertIs(a, b) a is b

assertIsNot(a, b) a is not b

assertIsNone(x) x is None

assertIsNotNone(x) x is not None

assertIn(a, b) a in b

assertNotIn(a, b) a not in b

assertIsInstance(a, b) isinstance(a, b)

assertNotIsInstance(a, b) not isinstance(a, b)

简单案例：

demo.py文件内容如下：

#!/usr/bin/python
# -*- coding: utf-8 -*-

def add(a, b):
    return a+b

def minus(a, b):
    return a-b

test_demo_class.py 文件内容：

#!/usr/bin/python
# -*- coding: utf-8 -*-

import unittest
from .demo import add,minus



class TestDemo(unittest.TestCase):
    """Test mathfuc.py"""

    @classmethod
    def setUpClass(cls):
        print ("this setupclass() method only called once.\n")

    @classmethod
    def tearDownClass(cls):
        print ("this teardownclass() method only called once too.\n")

    def setUp(self):
        print ("do something before test : prepare environment.\n")

    def tearDown(self):
        print ("do something after test : clean up.\n")

    def test_add(self):
        """Test method add(a, b)"""
        self.assertEqual(3, add(1, 2))
        self.assertNotEqual(4, add(2, 2))

    def test_minus(self):
        """Test method minus(a, b)"""
        self.assertEqual(1, minus(3, 2))
        self.assertNotEqual(1, minus(3, 2),"hahahahah")
	#下注解表示该用例未写好，先不测
    @unittest.skip("do't run as not ready")
    def test_minus_with_skip(self):
        """Test method minus(a, b)"""
        self.assertEqual(1, minus(3, 2))
        self.assertNotEqual(1, minus(3, 2))


if __name__ == '__main__':
    # verbosity=*：默认是1；设为0，则不输出每一个用例的执行结果；2-输出详细的执行结果
    unittest.main()

执行结果：

Testing started at 17:54 ...
/Users/wh/Documents/0003-项目管理/024-smart-iot/venv/bin/python /Applications/PyCharm.app/Contents/plugins/python/helpers/pycharm/_jb_unittest_runner.py --target test.test_demo_class.TestDemo
Launching unittests with arguments python -m unittest test.test_demo_class.TestDemo in /Users/wh/Documents/0003-项目管理/024-smart-iot/git@gitlab.raysdata-lab.com:smart-iot/thingsboard-gateway-python

this setupclass() method only called once.

do something before test : prepare environment.

do something after test : clean up.


Failure
Traceback (most recent call last):
  File "/Users/wh/.pyenv/versions/3.7.5/lib/python3.7/unittest/case.py", line 59, in testPartExecutor
    yield
  File "/Users/wh/.pyenv/versions/3.7.5/lib/python3.7/unittest/case.py", line 628, in run
    testMethod()
  File "/Users/wh/Documents/0003-项目管理/024-smart-iot/git@gitlab.raysdata-lab.com:smart-iot/thingsboard-gateway-python/test/test_demo_class.py", line 29, in test_add
    self.assertNotEqual(4, add(2, 2))
  File "/Users/wh/.pyenv/versions/3.7.5/lib/python3.7/unittest/case.py", line 861, in assertNotEqual
    raise self.failureException(msg)
AssertionError: 4 == 4

do something before test : prepare environment.

do something after test : clean up.


Failure
Traceback (most recent call last):
  File "/Users/wh/.pyenv/versions/3.7.5/lib/python3.7/unittest/case.py", line 59, in testPartExecutor
    yield
  File "/Users/wh/.pyenv/versions/3.7.5/lib/python3.7/unittest/case.py", line 628, in run
    testMethod()
  File "/Users/wh/Documents/0003-项目管理/024-smart-iot/git@gitlab.raysdata-lab.com:smart-iot/thingsboard-gateway-python/test/test_demo_class.py", line 34, in test_minus
    self.assertNotEqual(1, minus(3, 2),"hahahahah")
  File "/Users/wh/.pyenv/versions/3.7.5/lib/python3.7/unittest/case.py", line 861, in assertNotEqual
    raise self.failureException(msg)
AssertionError: 1 == 1 : hahahahah


Skipped: do't run as not ready

Assertion failed

Assertion failed


Ran 3 tests in 0.005s
this teardownclass() method only called once too.


FAILED (failures=2, skipped=1)

Process finished with exit code 1

Assertion failed

Assertion failed

类TestDemo继承自unittest.TestCase，在类中定义了3个testcase。

unittest.main()调用Testloader加载这3个testcase到Testsuite，再用Testrunner运行testsuite，生成Testresult。

从output中可以看出，执行每个testcase之前都会先setUp()初始化，执行完毕之后tearDown()清理环境。

四、unittest主要常用函数

官方用例
在这里插入图片描述
python的unittest模块提供了一个测试框架，只要我们写一个继承unittest.TestCase的类，类中用setUp做初始化，用tearDown做清理。

主要用到的函数有：

failedinfo表示不成立打印信息failedinfo，为可选参数

self.fail([msg])会无条件的导致测试失败，不推荐使用。

self.assertEqual(value1, value2, failedinfo) # 断言value1 == value2

self.assertTrue(表达式, failedinfo) # 断言value为真

self.assertFalse(表达式, failedinfo) # 断言value为假
断言肯定发生异常，如果没发生异常，则为测试失败。
参数1为异常，参数二为抛出异常的调用对象，剩余参数为传递给可调用对象的参数。
self.assertRaises(ValueError, self.widget.resize, -1, -1)

调用时机的加self,如self.assertEqual(self.seq, range(10))，self.assertTrue(value > 100)

五、unittest网易邮箱登录案例

1.打开网易邮箱，写一个简单的登录；

2.判断title完全等于期望结果；
在这里插入图片描述

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author  : Ailie
# @File    : mailLogin.py
# @Software: PyCharmimport timeimport unittestfrom selenium 
import webdriverfrom selenium.webdriver.support 
import expected_conditions as ECclass mailLogin(unittest.TestCase):
    def setUp(self):
        url = 'https://mail.yeah.net/'
        self.browser = webdriver.Firefox()
        self.browser.get(url)
        time.sleep(5)    def test_login_01(self):
        '''
        用户名、密码为空
        '''
        self.browser.switch_to.frame("x-URS-iframe")
        self.browser.find_element_by_name('email').send_keys('')
        self.browser.find_element_by_name('password').send_keys('')
        self.browser.find_element_by_id('dologin').click()
        self.browser.switch_to.default_content()
        time.sleep(3)
        name = self.browser.find_element_by_id('spnUid')        if name == 'sanzang520@yeah.net':
            print('登录成功')        else:
            print('登陆失败')    def test_login_02(self):
        '''
        用户名正确、密码为错误
        '''
        self.browser.switch_to.frame("x-URS-iframe")
        self.browser.find_element_by_name('email').send_keys('sanzang520')
        self.browser.find_element_by_name('password').send_keys('xxx')
        self.browser.find_element_by_id('dologin').click()
        self.browser.switch_to.default_content()
        time.sleep(3)
        name = self.browser.find_element_by_id('spnUid')        if name == 'sanzang520@yeah.net':
            print('登录成功')        else:
            print('登陆失败')    def test_login_03(self):
        '''
        用户名、密码正确
        '''
        self.browser.switch_to.frame("x-URS-iframe")
        self.browser.find_element_by_name('email').send_keys('sanzang520')
        self.browser.find_element_by_name('password').send_keys('xxx')
        self.browser.find_element_by_id('dologin').click()
        self.browser.switch_to.default_content()
        time.sleep(3)
        name = self.browser.find_element_by_id('spnUid')        if name == 'sanzang520@yeah.net':
            print('登录成功')        else:
            print('登陆失败')    def tearDown(self):
        self.browser.quit()if __name__ == "__main__":
    unittest.main()

杯子很干净

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
unittest简单上手

一、什么是unittestunittest是Python单元测试框架，类似于JUnit框架。unittest中有4个重要的概念：test fixture, test case, test suite, test runnerTestcase：一个TestCase的实例就是一个测试用例。什么是测试用例呢？就是一个完整的测试流程，包括测试前准备环境的搭建(setUp)，执行测试代码 (run)，以及测试后环境的还原(tearDown)。元测试(unit test)的本质也就在这里，一个测试用例是一个完整
复制链接

扫一扫