Python学习打call第七十三天:unittest断言的使用

断言使用

  • 我们可以直接在测试用例里面添加unittest框架自带的断言(也就相当于判断),断言方法提供了测试结果是True还是False,所有的断言方法都有一个msg参数,如果指定msg参数的值,则将该信息作为失败的错误信息返回;

  • 如果断言失败则抛出一个AssertionError,并标识该测试为失败状态;如果成功,则标识该测试为成功状态;如果异常,则当做错误来处理;

先来看一个实例,在一个测试类MyTestCase中添加多个测试方法和断言

from unittest import TestCase, main, skip
class MyTestCase(TestCase):
    def setUp(self):
        print('setUp')
    def tearDown(self):
        print('tearDown')
    def test_print1(self):
        print('第一个测试用例')
        # unittest框架自带的断言,如果不相等,直接抛出异常,相等什么都不做
        self.assertEqual(200, 200, '测试失败')
    def test_print2(self):
        print('第二个测试用例')
    def test_print3(self):
        print('第三个测试用例')
if __name__ == '__main__':
    main()

========输出========:
setUp
第一个测试用例
tearDown
setUp
第二个测试用例
tearDown
setUp
第三个测试用例
tearDown

这里有三个测试用例,每个测试用例开始之前都先执行了setUp方法,每个测试用例之后都执行了tearDown方法。如果我们想要忽略某个测试用例不执行,要怎么做?比如忽略第三个测试用例,只需要在第三个测试用例之前加上skip装饰器:

@skip('do not need test_print3')
    def test_print3(self):
        print('第三个测试用例')

skip装饰器

  • 我们可以使用unitest.skip装饰器跳过不需要的测试用例,@unittest.skip(reason)表示无条件跳过测试用例,reason是用来说明原因的;

  • @unittest.skipIf(conditition,reason):condititon为true时跳过测试用例;

  • @unittest.skipunless(condition,reason):condition为False时跳过测试用例;

之前我们学习了setup方法和setUpClass方法,会在每个测试用例的前后都执行一遍,那如果我们只想要在所有测试用例的前后执行一遍,也就是在整个类中只执行一遍要怎么办呢,我们可以使用setUpClass方法和tearDownClass方法:

from unittest import TestCase, main, skip
class MyTestCase(TestCase):
    @classmethod
    def setUpClass(cls):
        print('setUpClass')
    @classmethod
    def tearDownClass(cls):
        print('tearDownClass')
    def test_print1(self):
        print('第一个测试用例')
        # unittest框架自带的断言,如果不相等,直接抛出异常,相等什么都不做
        self.assertEqual(200, 200, '测试失败')
    def test_print2(self):
        print('第二个测试用例')

    @skip('do not need test_print3')
    def test_print3(self):
        print('第三个测试用例')
if __name__ == '__main__':
    main()


==========输出===========
setUpClass
第一个测试用例
第二个测试用例
tearDownClass

setUpClass和tearDownClass的使用

  • setup方法和teardown方法在每次执行一个测试用例时都会重新执行一遍,所以要想在整个类中只执行一遍可以用setUpClass和tearDownClass方法;

  • setUpClass:需要添加@classmethod 装饰器,在一个类中只会在所有测试用例运行前调用一次;

  • tearDownClass:需要添加@classmethod 装饰器,在所有测试用例运行后调用一次;

引入测试用例集,将测试用例与测试集合分开

  • 首先可以先在一个文件中写好测试用例模块,然后在另一个文件中导入这个模块,再使用unittest.TestSuite类创建测试套件,然后使用这个类的addTest() 方法将需要被测试的测试用例添加到测试套件中,最后使用unittest.TextTestRunner类中的run()方法来运行suite所组装的测试用例;
  • 测试用例模块
from unittest import TestCase
from unittest import skip
from unittest import main as main_method
class MyTestCase(TestCase):
    @classmethod
    def setUpClass(cls):
        print('setUpClass')
    @classmethod
    def tearDownClass(cls):
        print('tearDownClass')
    def test_print1(self):
        print('第一个测试用例')
        self.assertEqual(200, 200, '测试失败')
    def test_print2(self):
        print('第一个测试用例')

    @skip('do not need test_print3')
    def test_print3(self):
        print('第一个测试用例')
if __name__ == '__main__':
    main_method()
  • 测试集模块
from test1 import MyTestCase
from unittest import TestSuite, TextTestRunner
suite = TestSuite()
# 在测试用例集中添加测试用例, 只添加两个测试的方法
suite.addTest(MyTestCase('test_print1'))
suite.addTest(MyTestCase('test_print3'))
if __name__ == '__main__':
    # 创建测试runner,执行测试用例集
    runner = TextTestRunner()
    runner.run(suite)

=========输出=============
setUpClass
第一个测试用例
tearDownClass

我们运行这段代码,只有第一个测试用例输出,因为我们只将第一个和第三个测试用例加入到测试集中,所有第二个测试用例不会执行,而我们在第三个测试用例前面机上了skip装饰器会跳过此测试用例,所有第三个也不会执行啦。

将待测试函数、测试用例、测试用例集分开

  • 将待测试函数、测试用例、测试用例集分别放在不同的文件中,先在测试用例中导入待测试函数,然后使用断言判断函数是否正确,接着在测试用例集中创健测试套件,且将测试用例加入套件里面,最后运行;
  • 测试函数
def add(x, y):
    return x + y
def sub(x, y):
    return x - y
  • 测试用例
from unittest import TestCase
from unittest import skip
from unittest import main as main_method
from add_sub import add, sub

class MyTestCase(TestCase):
    @classmethod
    def setUpClass(cls):
        print('setUpClass')
    @classmethod
    def tearDownClass(cls):
        print('tearDownClass')
    def test_print1(self):
        ret = 1 + 2
        result = add(1, 2)
        self.assertEqual(result, ret, 'test_print1测试失败')

    def test_print2(self):
        print('第二个测试用例')

    @skip('do not need test_print3')
    def test_print3(self):
        print('第三个测试用例')
        
if __name__ == '__main__':
    main_method()
  • 测试用例集
from test import MyTestCase
from unittest import TestSuite, TextTestRunner
suite = TestSuite()
# 在测试用例集中添加测试用例
suite.addTest(MyTestCase('test_print1'))
suite.addTest(MyTestCase('test_print2'))
suite.addTest(MyTestCase('test_print3'))

if __name__ == '__main__':
    
    # 创建测试runner,执行测试用例集
    runner = TextTestRunner()
    runner.run(suite)

========输出=========
setUpClass
第一个测试用例
第二个测试用例
tearDownClass

参考:https://www.9xkd.com/user/plan-view.html?id=4822012440

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,864评论 6 494
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,175评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,401评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,170评论 1 286
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,276评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,364评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,401评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,179评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,604评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,902评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,070评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,751评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,380评论 3 319
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,077评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,312评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,924评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,957评论 2 351

推荐阅读更多精彩内容