Python下测试框架unittest简单使用

最新推荐文章于 2022-09-04 12:22:17 发布

原创最新推荐文章于 2022-09-04 12:22:17 发布 · 546 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #unittest

python 专栏收录该内容

2 篇文章

订阅专栏

本文介绍了如何在Python中使用unittest框架进行单元测试。从待测模块开始，讲解了编写测试用例的步骤，包括导入unittest库，定义测试用例类，并说明测试方法应以test开头。最后提到了main函数的使用和测试结果的查看。

一、待测模块

class MyFunc:
    def __init__(self,number,a,b):
        self.number = number
        self.a = a
        self.b = b

    def is_prime(slef):
        if slef.number < 0 or slef.number in (0, 1):
            return False
        for element in range(2, slef.number):
            if slef.number % element == 0:
                return False
        return True

    def add(slef):
        return slef.a + slef.b

    def divide(slef):
        return slef.a / slef.b

二、编写测试用例
首先需要导入unitest框架和待测模块myfunc，定义的测试用例方法类，需要继承unittest.TestCase,且测试用例方法是以test开头作为标识

class TestMyFunc(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        print(" do once before all test")

    @classmethod
    def tearDownClass(cls):
        print(" do once before all test")

    def setUp(self):
        print("do before every test")

    def tearDown(self):
        print("do after every test")

    # 测试用例以test作为开头
    def test_is_prime(self):
        print("test_is_prime begin")
        func = MyFunc(5,2,3)
        self.assertTrue(func.is_prime())
        print("test_is_prime end")

    def test_add(self):
        print("test_add begin")
        func = MyFunc(5,2,3)
        self.assertEqual(5,func.add())
       #  self.assertEqual(6,func.add())
        print("test_add end")

    def test_divide(self):
        print("test_divide begin")
        func = MyFunc(5, 9, 3)
        self.assertEqual(3, func.divide())
        # self.assertEqual(2, func.divide())

三、mian函数

if __name__ == '__main__':
    unittest.main()
# 测试用例执行顺序为 测试用例名称升序
# 执行顺序为 test_add test_divide test_is_prime

四、测试结果
在这里插入图片描述