eth2.0-specs/test_libs/gen_helpers/gen_from_tests/gen.py

from inspect import getmembers, isfunction

def generate_from_tests(src, phase, 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 phase: to run tests against particular phase.
    :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:
            test_case = tfn(generator_mode=True, phase=phase, bls_active=bls_active)
            # If no test case data is returned, the test is ignored.
            if test_case is not None:
                out.append(test_case)
        except AssertionError:
            print("ERROR: failed to generate vector from test: %s (src: %s)" % (name, src.__name__))
    return out