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

[guilherme.polo]

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