23 lines
891 B
Python
23 lines
891 B
Python
|
from inspect import getmembers, isfunction
|
||
|
|
|
||
|
|
def generate_from_tests(src, bls_active=True):
|
||
|
|
"""
|
||
|
|
Generate a list of test cases by running tests from the given src in generator-mode.
|
||
|
|
:param src: to retrieve tests from (discovered using inspect.getmembers)
|
||
|
|
:param bls_active: optional, to override BLS switch preference. Defaults to True.
|
||
|
|
:return: the list of test cases.
|
||
|
|
"""
|
||
|
|
fn_names = [
|
||
|
|
name for (name, _) in getmembers(src, isfunction)
|
||
|
|
if name.startswith('test_')
|
||
|
|
]
|
||
|
|
out = []
|
||
|
|
print("generating test vectors from tests source: %s" % src.__name__)
|
||
|
|
for name in fn_names:
|
||
|
|
tfn = getattr(src, name)
|
||
|
|
try:
|
||
|
|
out.append(tfn(generator_mode=True, bls_active=bls_active))
|
||
|
|
except AssertionError:
|
||
|
|
print("ERROR: failed to generate vector from test: %s (src: %s)" % (name, src.__name__))
|
||
|
|
return out
|