在之前的生活中,我做了很多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)
但如果第一个"案例"这是相当笨重的。失败了,所有其他人都无法运行。
答案 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内置了类似内置的内容 - subTest
:https://docs.python.org/3.4/library/unittest.html#distinguishing-test-iterations-using-subtests
不如py.test的参数化测试或jUnit Theories那么优雅,但如果你想要一个标准的lib方法&正在使用相对较新版本的Python,这是一个选项。