用于参数元组测试的Pytest文本标注
问题描述
我正在为这类问题寻找更优雅的解决方案:
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函数类实现的内部细节,因此不是很健壮。
我为每个参数添加了带有格式字符串和转换器的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
相关文章