76
76
class TestLoader(unittest.TestLoader):
77
"""Custom TestLoader to address some quirks in the stock python one."""
77
"""Custom TestLoader to extend the stock python one."""
78
79
suiteClass = TestSuite
80
81
def loadTestsFromModuleNames(self, names):
87
88
result = self.suiteClass()
89
_load_module_by_name(name)
90
result.addTests(self.loadTestsFromName(name))
90
module = _load_module_by_name(name)
91
result.addTests(self.loadTestsFromModule(module))
94
def loadTestsFromModule(self, module):
95
"""Load tests from a module object.
97
This extension of the python test loader looks for an attribute
98
load_tests in the module object, and if not found falls back to the
99
regular python loadTestsFromModule.
101
If a load_tests attribute is found, it is called and the result is
104
load_tests should be defined like so:
105
>>> def load_tests(standard_tests, module, loader):
108
standard_tests is the tests found by the stock TestLoader in the
109
module, module and loader are the module and loader instances.
111
For instance, to run every test twice, you might do:
112
>>> def load_tests(standard_tests, module, loader):
113
>>> result = loader.suiteClass()
114
>>> for test in iter_suite_tests(standard_tests):
115
>>> result.addTests([test, test])
118
basic_tests = super(TestLoader, self).loadTestsFromModule(module)
119
load_tests = getattr(module, "load_tests", None)
120
if load_tests is not None:
121
return load_tests(basic_tests, module, self)
94
126
def _load_module_by_name(mod_name):
95
127
parts = mod_name.split('.')
added
removed
bzrlib/tests/TestUtil.py
