29

可能重复：
如何在 python 中生成动态（参数化）单元测试？

我正在使用 unittest 包编写测试，我想避免重复代码。我将进行许多测试，它们都需要非常相似的方法，但每次只有一个值不同。一个简单而无用的例子是：

class ExampleTestCase(unittest.TestCase):

    def test_1(self):
        self.assertEqual(self.somevalue, 1)

    def test_2(self):
        self.assertEqual(self.somevalue, 2)

    def test_3(self):
        self.assertEqual(self.somevalue, 3)

    def test_4(self):
        self.assertEqual(self.somevalue, 4)

有没有办法写上面的例子，而不是每次都重复所有的代码，而是写一个通用的方法，例如

    def test_n(self, n):
        self.assertEqual(self.somevalue, n)

并告诉 unittest 用不同的输入来尝试这个测试？

4

6 回答 6

11

一些可用于在 Python 中进行参数化测试的工具是：

于 2012-10-05T06:44:23.647 回答
3

如果你真的想要多个 unitttest 那么你需要多个方法。获得它的唯一方法是通过某种代码生成。您可以通过元类来做到这一点，或者在定义之后调整类，包括（如果您使用的是 Python 2.6）通过类装饰器。

这是一个寻找特殊的'multitest'和'multitest_values'成员并使用它们动态构建测试方法的解决方案。不优雅，但它大致可以满足您的要求：

import unittest
import inspect

class SomeValue(object):
    def __eq__(self, other):
        return other in [1, 3, 4]

class ExampleTestCase(unittest.TestCase):
    somevalue = SomeValue()

    multitest_values = [1, 2, 3, 4]
    def multitest(self, n):
        self.assertEqual(self.somevalue, n)

    multitest_gt_values = "ABCDEF"
    def multitest_gt(self, c):
        self.assertTrue(c > "B", c)


def add_test_cases(cls):
    values = {}
    functions = {}
    # Find all the 'multitest*' functions and
    # matching list of test values.
    for key, value in inspect.getmembers(cls):
        if key.startswith("multitest"):
            if key.endswith("_values"):
                values[key[:-7]] = value
            else:
                functions[key] = value

    # Put them together to make a list of new test functions.
    # One test function for each value
    for key in functions:
        if key in values:
            function = functions[key]
            for i, value in enumerate(values[key]):
                def test_function(self, function=function, value=value):
                    function(self, value)
                name ="test%s_%d" % (key[9:], i+1)
                test_function.__name__ = name
                setattr(cls, name, test_function)

add_test_cases(ExampleTestCase)

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

这是我运行它时的输出

% python stackoverflow.py
.F..FF....
======================================================================
FAIL: test_2 (__main__.ExampleTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "stackoverflow.py", line 34, in test_function
    function(self, value)
  File "stackoverflow.py", line 13, in multitest
    self.assertEqual(self.somevalue, n)
AssertionError: <__main__.SomeValue object at 0xd9870> != 2

======================================================================
FAIL: test_gt_1 (__main__.ExampleTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "stackoverflow.py", line 34, in test_function
    function(self, value)
  File "stackoverflow.py", line 17, in multitest_gt
    self.assertTrue(c > "B", c)
AssertionError: A

======================================================================
FAIL: test_gt_2 (__main__.ExampleTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "stackoverflow.py", line 34, in test_function
    function(self, value)
  File "stackoverflow.py", line 17, in multitest_gt
    self.assertTrue(c > "B", c)
AssertionError: B

----------------------------------------------------------------------
Ran 10 tests in 0.001s

FAILED (failures=3)

您可以立即看到代码生成中出现的一些问题。“test_gt_1”从何而来？我可以将名称更改为更长的“test_multitest_gt_1”，但是哪个测试是 1？最好从 _0 而不是 _1 开始，也许在您的情况下，您知道这些值可以用作 Python 函数名称。

我不喜欢这种方法。我研究过自动生成测试方法的代码库（在一种情况下使用元类），发现它更难理解而不是有用。当测试失败时，很难找出失败案例的来源，也很难坚持调试代码来探查失败的原因。

（我在这里编写的示例中的调试失败并不像我必须使用的特定元类方法那么难。）

于 2009-11-05T11:56:33.177 回答
1

一种更面向数据的方法可能比Andrew Dalke的回答中使用的方法更清晰：

"""Parametrized unit test.

Builds a single TestCase class which tests if its
  `somevalue` method is equal to the numbers 1 through 4.

This is accomplished by
  creating a list (`cases`)
  of dictionaries which contain test specifications
  and then feeding the list to a function which creates a test case class.

When run, the output shows that three of the four cases fail,
  as expected:

>>> import sys
>>> from unittest import TextTestRunner
>>> run_tests(TextTestRunner(stream=sys.stdout, verbosity=9))
... # doctest: +ELLIPSIS
Test if self.somevalue equals 4 ... FAIL
Test if self.somevalue equals 1 ... FAIL
Test if self.somevalue equals 3 ... FAIL
Test if self.somevalue equals 2 ... ok
<BLANKLINE>
======================================================================
FAIL: Test if self.somevalue equals 4
----------------------------------------------------------------------
Traceback (most recent call last):
  ...
AssertionError: 2 != 4
<BLANKLINE>
======================================================================
FAIL: Test if self.somevalue equals 1
----------------------------------------------------------------------
Traceback (most recent call last):
  ...
AssertionError: 2 != 1
<BLANKLINE>
======================================================================
FAIL: Test if self.somevalue equals 3
----------------------------------------------------------------------
Traceback (most recent call last):
  ...
AssertionError: 2 != 3
<BLANKLINE>
----------------------------------------------------------------------
Ran 4 tests in ...s
<BLANKLINE>
FAILED (failures=3)
"""

from unittest import TestCase, TestSuite, defaultTestLoader

cases = [{'name': "somevalue_equals_one",
          'doc': "Test if self.somevalue equals 1",
          'value': 1},
         {'name': "somevalue_equals_two",
          'doc': "Test if self.somevalue equals 2",
          'value': 2},
         {'name': "somevalue_equals_three",
          'doc': "Test if self.somevalue equals 3",
          'value': 3},
         {'name': "somevalue_equals_four",
          'doc': "Test if self.somevalue equals 4",
          'value': 4}]

class BaseTestCase(TestCase):
    def setUp(self):
        self.somevalue = 2

def test_n(self, n):
    self.assertEqual(self.somevalue, n)

def make_parametrized_testcase(class_name, base_classes, test_method, cases):
    def make_parametrized_test_method(name, value, doc=None):
        def method(self):
            return test_method(self, value)
        method.__name__ = "test_" + name
        method.__doc__ = doc
        return (method.__name__, method)

    test_methods = (make_parametrized_test_method(**case) for case in cases)
    class_dict = dict(test_methods)
    return type(class_name, base_classes, class_dict)


TestCase = make_parametrized_testcase('TestOneThroughFour',
                                      (BaseTestCase,),
                                      test_n,
                                      cases)

def make_test_suite():
    load = defaultTestLoader.loadTestsFromTestCase
    return TestSuite(load(TestCase))

def run_tests(runner):
    runner.run(make_test_suite())

if __name__ == '__main__':
    from unittest import TextTestRunner
    run_tests(TextTestRunner(verbosity=9))

我不确定在确定测试运行顺序时涉及什么巫术，但至少对我来说，doctest 始终如一地通过。

对于更复杂的情况，可以将字典的values元素替换为cases包含参数列表和关键字参数字典的元组。尽管那时您基本上是在 python 中编写 lisp 代码。

于 2010-08-18T22:13:40.337 回答
1

我猜你想要的是“参数化测试”。

我不认为 unittest 模块支持这个（不幸的是），但如果我添加这个功能，它看起来像这样：

# Will run the test for all combinations of parameters
@RunTestWith(x=[0, 1, 2, 3], y=[-1, 0, 1])
def testMultiplication(self, x, y):
  self.assertEqual(multiplication.multiply(x, y), x*y)

使用现有的 unittest 模块，像这样的简单装饰器将无法多次“复制”测试，但我认为使用装饰器和元类的组合是可行的（元类应该遵守所有 'test*' 方法并复制（在不同的自动生成名称下）应用了装饰器的那些）。

于 2009-11-04T22:46:01.107 回答
0

编写一个测试方法来执行所有测试并捕获所有结果，将您自己的诊断消息写入 stderr，如果任何子测试失败，则测试失败：

def test_with_multiple_parameters(self):
    failed = False
    for k in sorted(self.test_parameters.keys()):
        if not self.my_test(self.test_parameters[k]):
           print >> sys.stderr, "Test {0} failed.".format(k)
           failed = True
    self.assertFalse(failed)

请注意，当然my_test()的名称不能以 . 开头test。

于 2009-11-05T15:12:44.833 回答
-1

也许是这样的：

def test_many(self):
    for n in range(0,1000):
        self.assertEqual(self.somevalue, n)
于 2009-11-04T20:09:18.797 回答