Python单元测试如何传参

在编写Python代码时，进行单元测试是保证代码质量的重要环节，我们需要在单元测试中传递参数以验证不同条件下的代码行为，如何在Python单元测试中传参呢？本文将详细介绍几种在Python单元测试中传参的方法。

我们需要了解Python中常用的单元测试框架，如unittest和pytest，这两个框架都支持在测试用例中传递参数，下面我们将分别介绍在这两个框架中如何实现传参。

使用unittest框架传参

在unittest框架中，我们可以通过覆写TestCase类的setUp方法来为每个测试用例设置参数，如果我们要传递的参数是固定的，这种方法可能不太适用，以下是一种使用unittest传参的常见方法：

1、使用构造函数传参

我们可以通过自定义测试用例的构造函数来接收参数，然后在测试方法中使用这些参数。

import unittest
class TestMyFunction(unittest.TestCase):
    def __init__(self, test_name, param1, param2):
        super().__init__(test_name)
        self.param1 = param1
        self.param2 = param2
    def test_my_function(self):
        result = my_function(self.param1, self.param2)
        self.assertEqual(result, expected_result)
if __name__ == '__main__':
    unittest.main()

2、使用addTest方法传参

我们可以使用TestSuite类的addTest方法动态地添加测试用例，并为每个测试用例传递不同的参数。

import unittest
class TestMyFunction(unittest.TestCase):
    def test_my_function(self, param1, param2):
        result = my_function(param1, param2)
        self.assertEqual(result, expected_result)
def suite():
    suite = unittest.TestSuite()
    suite.addTest(TestMyFunction('test_my_function', 1, 2))
    suite.addTest(TestMyFunction('test_my_function', 3, 4))
    return suite
if __name__ == '__main__':
    unittest.main(defaultTest='suite')

使用pytest框架传参

pytest是一个更强大、更易于使用的单元测试框架，它支持多种传参方式。

1、使用装饰器传参

Python单元测试如何传参

pytest提供了@pytest.mark.parametrize装饰器，可以很方便地为测试用例传递参数。

import pytest
@pytest.mark.parametrize("param1,param2", [(1, 2), (3, 4)])
def test_my_function(param1, param2):
    result = my_function(param1, param2)
    assert result == expected_result

2、使用测试用例生成器传参

我们还可以使用生成器函数来创建多个测试用例，并为每个用例传递不同的参数。

import pytest
def gen_params():
    for i in range(1, 5):
        for j in range(1, 5):
            yield (i, j)
@pytest.mark.parametrize("param1,param2", gen_params())
def test_my_function(param1, param2):
    result = my_function(param1, param2)
    assert result == expected_result