Python中的单元测试理论?

时间:2015-05-27 22:49:14

标签: python unit-testing junit-theory

在之前的生活中,我做了很多Java开发,发现JUnit Theories非常有用。 Python有没有类似的机制?

目前,我正在做类似的事情:

def some_test(self):
    cases = [('some sample input', 'some expected value'),
            ('some other input', 'some other expected value')]

    for value, expected in cases:
        result = method_under_test(value)
        self.assertEqual(expected, result)

但如果第一个"案例"这是相当笨重的。失败了,所有其他人都无法运行。

3 个答案:

答案 0 :(得分:2)

看起来pytest可以做这样的事情:http://pytest.org/latest/parametrize.html#parametrize-basics

我自己没有尝试过。

答案 1 :(得分:0)

我在任何常见的测试框架中都不知道有任何内置函数。解决方案的唯一问题是迭代是在测试内部。相反它应该在外面并生成测试,可能是这样的

import unittest


def _apply(func, args):
    """Return a function with args applied after first argument"""
    def wrapped(self):
        return func(self, *args)
    return wrapped


class TheoryMeta(type):
    """Metaclass that replaces test methods with multiple methods for each test case"""
    def __new__(meta, name, bases, attrs):
        newattrs = {}
        cases = attrs.pop('cases', [])

        for name, value in attrs.items():
            if not name.startswith('test') or not callable(value):
                newattrs[name] = value
                continue

            for n, args in enumerate(cases):
                test_name = '%s_%d' % (name, n)
                newattrs[test_name] = _apply(value, args)

        return super().__new__(meta, name, bases, newattrs)


class TestCase(unittest.TestCase, metaclass=TheoryMeta):
    pass

然后使用它,创建一个TestCase子类,该子类具有cases属性,该属性是应用于测试用例上每个测试方法的参数列表。

class TestAdd(TestCase):
    cases = [
        # (a, b)
        (1, 1),
        (2, 0),
        (3, 0),
    ]

    def test_add(self, a, b):
        self.assertEqual(a + b, 2)

======================================================================
FAIL: test_add_2 (__main__.__qualname__)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "test.py", line 7, in wrapped
    return func(self, *args)
  File "test.py", line 41, in test_add
    self.assertEqual(a + b, 2)
AssertionError: 3 != 2

----------------------------------------------------------------------
Ran 3 tests in 0.001s

根据您的需求和测试设置,您可能会发现在TestCase上修改生成的测试方法而不是使用元类更好。或者您可以在TestLoader上覆盖loadTestsFrom...生成它们。无论如何,请使用示例数据生成测试方法。

答案 2 :(得分:0)

事实证明,Python 3.4内置了类似内置的内容 - subTesthttps://docs.python.org/3.4/library/unittest.html#distinguishing-test-iterations-using-subtests

不如py.test的参数化测试或jUnit Theories那么优雅,但如果你想要一个标准的lib方法&正在使用相对较新版本的Python,这是一个选项。