cryptography/tests/wycheproof/utils.py
Alex Gaynor d844fd4556
Split wycheproof tests up by file (#10025)
This provides greater parallelism. 25% faster at running wycheproof tests locally
2023-12-21 12:16:49 -06:00

23 lines
752 B
Python

# This file is dual licensed under the terms of the Apache License, Version
# 2.0, and the BSD License. See the LICENSE file in the root of this repository
# for complete details.
import pytest
from ..utils import load_wycheproof_tests
def wycheproof_tests(*paths, subdir="testvectors"):
def wrapper(func):
@pytest.mark.parametrize("path", paths)
def run_wycheproof(backend, subtests, pytestconfig, path):
wycheproof_root = pytestconfig.getoption(
"--wycheproof-root", skip=True
)
for test in load_wycheproof_tests(wycheproof_root, path, subdir):
with subtests.test():
func(backend, test)
return run_wycheproof
return wrapper