unittest 简单使用-优快云博客

本文深入探讨了Python中unittest模块的使用方法，包括setUpClass、tearDownClass、setUp、tearDown等方法的作用，以及如何运行测试用例。同时，文章还介绍了三种执行测试用例的方法，为读者提供了详细的Python单元测试指南。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

2019独角兽企业重金招聘Python工程师标准>>>

import unittest


class MyTest(unittest.TestCase):

    # 使用类时调用，只调用一次
    @classmethod
    def setUpClass(cls):
        print("setup class")

    # 使用类结束时调用，只调用一次
    @classmethod
    def tearDownClass(cls):
        print("tear down class")

    # 用例执行前调用，每个测试方法都会调用
    def setUp(self):
        print("------> setup")

    def test_case1(self):
        print("------> test_case1")
        self.assertTrue(1 == 1)

    def test_case2(self):
        print("------> test_case2")
        self.assertEqual(1, 2)

    # skip标记的用例不会执行
    @unittest.skip
    def test_case3(self):
        print("------> test_case3")

    # 用例执行后调用，每个测试方法都会调用
    def tearDown(self):
        print("------>teardown")


if __name__ == '__main__':

    # 执行测试用例
    # 方法一：执行所有以test开头的用例
    unittest.main()

    # 方法二：使用TestSuite测试集添加需要执行的测试用例
    suite = unittest.TestSuite
    suite.addTest(MyTest("test_case1"))
    suite.addTest(MyTest("test_case2"))

    runner = unittest.TextTestRunner
    runner.run(suite)

    # 方法三：defaultTestLoader查找指定目录下、指定格式的Python文件
    discover = unittest.defaultTestLoader.discover("./", pattern="test*.py")
    runner = unittest.TextTestRunner
    runner.run(discover)

转载于:https://my.oschina.net/mexiaobai1315/blog/3021884