tests: Optimize run-tests.py test case dict iteration

Use a single iteration through the module dictionary rather than
iterating over the keys and separately fetching values to get the
function.

Signed-off-by: Jouni Malinen <j@w1.fi>
This commit is contained in:
Jouni Malinen 2014-12-22 20:08:11 +02:00
parent 4cfe369b98
commit 6ac593ec47

View file

@ -161,10 +161,9 @@ def main():
logger.debug("Import test cases from " + t) logger.debug("Import test cases from " + t)
mod = __import__(m.group(1)) mod = __import__(m.group(1))
test_modules.append(mod.__name__.replace('test_', '', 1)) test_modules.append(mod.__name__.replace('test_', '', 1))
for s in dir(mod): for key,val in mod.__dict__.iteritems():
if s.startswith("test_"): if key.startswith("test_"):
func = mod.__dict__.get(s) tests.append(val)
tests.append(func)
test_names = list(set([t.__name__.replace('test_', '', 1) for t in tests])) test_names = list(set([t.__name__.replace('test_', '', 1) for t in tests]))
run = None run = None