python 单元测试 unittest

阅读目录

• 例1：测试函数
• 例2：测试类
• 例3：构造测试集

单元测试可以帮助我们准确迅速地定位出现问题的模块和单元。

单元测试的对象是函数，必须有输出结果（哪怕是异常输出），以便单元测试模块能够捕获返回值，并且与预期值进行比较，从而得出测试通过与否

单元测试的加载可以通过unittest.main()来启动单元测试的测试模块。

而testsuit里存放的就是单元测试的用例，下面分别列出了2种方法的使用。

• setUp():             每个测试函数运行前运行
• tearDown():        每个测试函数运行完后执行
• setUpClass():      必须使用@classmethod 装饰器,所有test运行前运行一次
• tearDownClass(): 必须使用@classmethod装饰器,所有test运行完后运行一次

注：引入模块   import unittest 

回到顶部

例1：测试函数

def sum(x, y):
    return x + y

def sub(x, y):
    return x - y

单元测试模块

class mytest(unittest.TestCase):   #所有执行测试的类都继承TestCase类
    #初始化工作
    def setUp(self):
        pass
    #清退工作
    def tearDown(self):
        pass
    #具体的测试用例，一定要以test开头
    def testsum_1(self):
        self.assertEqual(sum(1, 2), 3, "test sum fail")

    def testsum_2(self):
        self.assertEqual(sum(1, 2), 2, "test sum fail")

    def testsub(self):
        self.assertEqual(sub(2, 1), 1, "test sub fail")

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

测试结果

1 2 3 4 5 6 7 8 9 10 11 12 13

..F ====================================================================== FAIL: testsum_ 2  (__main__.mytest) ---------------------------------------------------------------------- Traceback (most recent call last):    File  "E:/task/��Ԫ����/test1.py" , line  18 , in testsum_ 2      self.assertEqual(sum( 1 ,  2 ),  2 ,  "test sum fail" ) AssertionError: test sum fail   ---------------------------------------------------------------------- Ran  3  tests in  0.003 s   FAILED (failures= 1 )

注：F表示fail， F前的点表示测试通过，有E的话表示程序自身异常

 

回到顶部

例2：测试类

class myclass:
    def __init__(self):
        pass

    def sum(self, x, y):
        return x + y

    def sub(self, x, y):
        return x - y

测试代码

class mytest(unittest.TestCase):
    def setUp(self):
        self.tclass = myclass()

    def tearDown(self):
        pass

    def testsum_1(self):
        self.assertEqual(self.tclass.sum(1, 2), 3, "test sum fail")

    def testsum_2(self):
        self.assertEqual(self.tclass.sum(1, 2), 3, "test sum fail")

    def testsub(self):
        self.assertEqual(self.tclass.sub(2, 1), 1, "test sub fail")

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

结果

1 2 3 4 5 6 7 8 9 10 11 12 13

..F ====================================================================== FAIL: testsum_ 2  (__main__.mytest) ---------------------------------------------------------------------- Traceback (most recent call last):    File  "E:/task/��Ԫ����/test2.py" , line  24 , in testsum_ 2      self.assertEqual(self.tclass.sum( 1 ,  2 ),  5 ,  "test sum fail" ) AssertionError: test sum fail   ---------------------------------------------------------------------- Ran  3  tests in  0.001 s   FAILED (failures= 1 )

 

回到顶部

例3：构造测试集

被测代码

class myclass:
    def __init__(self):
        pass

    def sum(self, x, y):
        return x + y

    def sub(self, x, y):
        return x - y

测试代码

class mytest(unittest.TestCase):
    def setUp(self):
        self.tclass = myclass()

    def tearDown(self):
        pass

    def testsum_1(self):
        self.assertEqual(self.tclass.sum(1, 2), 3, "test sum fail")

    def testsum_2(self):
        self.assertEqual(self.tclass.sum(1, 2), 5, "test sum fail")

    def testsub(self):
        self.assertEqual(self.tclass.sub(2, 1), 1, "test sub fail")

def suite():
    suite = unittest.TestSuite()
    suite.addTest(mytest("testsum_1"))
    suite.addTest(mytest("testsum_2"))
    suite.addTest(mytest("testsub"))
    return suite

if __name__ == "__main__":
    unittest.main(defaultTest = "suite")

 

本文转自jihite博客园博客，原文链接：http://www.cnblogs.com/kaituorensheng/p/4447307.html，如需转载请自行联系原作者