[英]Pytest text annotation for test with tuple of parameters
我正在为这种问题寻找更优雅的解决方案:
def ids(x):
if isinstance(x, int):
return str(x)
elif isinstance(x, str):
return x[0]
@pytest.mark.parametrize("number, string",
[
(1, "One"),
(2, "Two"),
(3, "Three")
],
ids=ids)
def test_stupid(number, string):
assert 0 == 1
该代码将产生测试名称:“ 1-O”,“ 2-T”,“ 3-T”。
问题是pytest对元组中的所有参数使用相同的函数,而我必须处理难看的isinstance调用。
有解决这个问题的更好方法吗?
我发现的唯一解决方案是使用钩子pytest_collection_modifyitems(session, config, items)
它可以访问参数值,并可用于更改测试名称。 但是它依赖于pytest Function类实现的内部细节,因此不是很可靠。
我为pytest.mark添加了格式字符串和每个参数的转换器。 所有参数可选
@pytest.mark.parametrize("number, string",
[
(1, "One"),
(2, "Two"),
(3, "Three")
])
@pytest.mark.params_format('number={number} string={string}',
number=lambda x: str(x),
string=lambda x: x[0])
def test_stupid(number, string):
然后添加到conftest.py
钩子实现中以使用该标记参数并添加格式化的字符串以进行测试
def pytest_collection_modifyitems(session, config, items):
for item in items:
if not _get_marker(item, 'parametrize'):
continue
format_params_marker = _get_marker(item, 'params_format')
if not format_params_marker:
continue
params = item.callspec.params.copy()
param_formatters = format_params_marker.kwargs
for param_name, presenter in param_formatters.iteritems():
params[param_name] = presenter(params[param_name])
params_names_ordered = [name for name in item.fixturenames if name in params]
default_format = '-'.join(['{{{}}}'.format(param_name) for param_name in params_names_ordered])
format_string = format_params_marker.args[0] if format_params_marker.args \
else default_format
item.name = '{func}[{params}]'.format(func=item.name.split('[')[0],
params=format_string.format(**params))
def _get_marker(item, marker):
return item.keywords._markers.get(marker, None)
结果看起来像这样
test_stupid[number=1 string=O] PASSED
test_stupid[number=2 string=T] PASSED
test_stupid[number=3 string=T] PASSED
传递一个字符串列表而不是一个可调用的ids
:
import pytest
PARAMS = [
(1, "One"),
(2, "Two"),
(3, "Three")
]
def my_id(number, string):
return '-'.join([str(number), string[0]])
@pytest.mark.parametrize("number, string",
PARAMS,
ids=[my_id(number, string) for number, string in PARAMS])
def test_stupid(number, string):
assert 0 == 1
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.