diff options
Diffstat (limited to 'test/py/test_repo.py')
| -rw-r--r-- | test/py/test_repo.py | 23 |
1 files changed, 19 insertions, 4 deletions
diff --git a/test/py/test_repo.py b/test/py/test_repo.py index 76aeef3..f451e8f 100644 --- a/test/py/test_repo.py +++ b/test/py/test_repo.py @@ -57,14 +57,29 @@ def _test_repo_internal(env, repo, numof_processes, runs_per_process): assert repo.run_output_matches(output), f"Output doesn't match: {output}" -@pytest.mark.parametrize('numof_clients,runs_per_client', - [(1, 1), (1, 2), (1, 10), (2, 1), (2, 10), (10, 1), (10, 10)]) +# Reference: https://github.com/pytest-dev/pytest/issues/3628 +# Automatic generation of readable test IDs. +def my_parametrize(names, values, ids=None, **kwargs): + _names = names.split(',') if isinstance(names, str) else names + if not ids: + if len(_names) == 1: + ids = [f'{names}={v}' for v in values] + else: + ids = [ + '-'.join(f'{k}={v}' for k, v in zip(_names, combination)) + for combination in values + ] + return pytest.mark.parametrize(names, values, ids=ids, **kwargs) + + +@my_parametrize('runs_per_client', [1, 5]) +@my_parametrize('numof_clients', [1, 5]) def test_repo(env, test_repo, numof_clients, runs_per_client): _test_repo_internal(env, test_repo, numof_clients, runs_per_client) @pytest.mark.stress -@pytest.mark.parametrize('numof_clients,runs_per_client', - [(1, 2000), (4, 500)]) +@my_parametrize(('numof_clients', 'runs_per_client'), + [(10, 50), (1, 2000), (4, 500)]) def test_repo_stress(env, test_repo, numof_clients, runs_per_client): _test_repo_internal(env, test_repo, numof_clients, runs_per_client) |