0
点赞
收藏
分享

微信扫一扫

Python单元测试-Unittest(四)

Python百事通 2022-03-26 阅读 87

Test suite的用途是将测试收集想要进行测试的测试用例,以便unittest执行这些选中的测试用例。

unittest.main()
通常情况下main函数其实是一个默认的test suite,它默认会收集所有模块的测试用例,然后执行它们。以如下测试用例为例,在执行单元测试时,由于已经在代码下方设置了main函数,那么运行命令可以简化为"python test_apple.py"。

import unittest
from apple import Apple

class AppleTestCases(unittest.TestCase):
    def setUp(self):
        self.apple = Apple()
        #raise Exception("This is an exception!!!")
    
    def test_default_apple_quantity(self):
        self.assertEqual(self.apple.quantity(),200,'default quantity is wrong.')
    
    def test_re_fill_quantity(self):
        self.apple.re_fill(500)
        self.assertEqual(self.apple.quantity(),500,'wrong quantity after re-fill.')
    
    def tearDown(self):
        print("The testing is done.")

if __name__ == '__main__':
    unittest.main()

定制化测试集合
在实际项目中,可能需要定制化测试集合的情况比较多,因为正常项目中一个模块(功能模块)的测试用例可能有很多,而有时只需要执行部分即可。
首先可以先定义一个test suite函数,具体如下所示:

def suite1():
    suite = unittest.TestSuite()
    suite.addTest(AppleTestCases('test_default_apple_quantity'))
    return suite

其次在执行这些测试用例时,也需要添加类似如下代码,去调用TestSuite实例。

if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite1())
举报

相关推荐

0 条评论