[Python-checkins] r69060 - python/trunk/Lib/lib-tk/test/runtktests.py

Wed Jan 28 20:23:28 CET 2009

Author: guilherme.polo
Date: Wed Jan 28 20:23:28 2009
New Revision: 69060
Log:
Added support for collecting tests only from specific packages.
Modified:
 python/trunk/Lib/lib-tk/test/runtktests.py
Modified: python/trunk/Lib/lib-tk/test/runtktests.py
==============================================================================

--- python/trunk/Lib/lib-tk/test/runtktests.py	(original)
+++ python/trunk/Lib/lib-tk/test/runtktests.py	Wed Jan 28 20:23:28 2009
@@ -19,9 +19,13 @@
 return True
 return False
 
-def get_tests_modules(basepath=this_dir_path, gui=True):
+def get_tests_modules(basepath=this_dir_path, gui=True, packages=None):
 """This will import and yield modules whose names start with test_
- and are inside packages found in the path starting at basepath."""
+ and are inside packages found in the path starting at basepath.
+
+ If packages is specified it should contain package names that want
+ their tests colleted.
+ """
 py_ext = '.py'
 
 for dirpath, dirnames, filenames in os.walk(basepath):
@@ -31,6 +35,9 @@
 
 if is_package(dirpath) and filenames:
 pkg_name = dirpath[len(basepath) + len(os.sep):].replace('/', '.')
+ if packages and pkg_name not in packages:
+ continue
+
 filenames = filter(
 lambda x: x.startswith('test_') and x.endswith(py_ext),
 filenames)
@@ -45,7 +52,7 @@
 if gui:
 raise
 
-def get_tests(text=True, gui=True):
+def get_tests(text=True, gui=True, packages=None):
 """Yield all the tests in the modules found by get_tests_modules.
 
 If nogui is True, only tests that do not require a GUI will be
@@ -55,7 +62,7 @@
 attrs.append('tests_nogui')
 if gui:
 attrs.append('tests_gui')
- for module in get_tests_modules(gui=gui):
+ for module in get_tests_modules(gui=gui, packages=packages):
 for attr in attrs:
 for test in getattr(module, attr, ()):
 yield test
