Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-10-31 13:41:24 +00:00
Code Issues Projects Releases Packages Wiki Activity

Patch #787789: allow to pass custom TestRunner instances to unittest's

Browse source 
main() function.
This commit is contained in:
Georg Brandl 2007-03-07 09:21:06 +00:00
parent 15c5ce936f
commit d0a962506b
3 changed files with 20 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

16
Lib/unittest.py
View file

@ -776,7 +776,8 @@ class TestProgram:
in MyTestCase
"""
def __init__(self, module='__main__', defaultTest=None,
argv=None, testRunner=None, testLoader=defaultTestLoader):
argv=None, testRunner=TextTestRunner,
testLoader=defaultTestLoader):
if type(module) == type(''):
self.module = __import__(module)
for part in module.split('.')[1:]:
@ -826,9 +827,16 @@ def createTests(self):
self.module)
def runTests(self):
if self.testRunner is None:
self.testRunner = TextTestRunner(verbosity=self.verbosity)
result = self.testRunner.run(self.test)
if isinstance(self.testRunner, (type, types.ClassType)):
try:
testRunner = self.testRunner(verbosity=self.verbosity)
except TypeError:
# didn't accept the verbosity argument
testRunner = self.testRunner()
else:
# it is assumed to be a TestRunner instance
testRunner = self.testRunner
result = testRunner.run(self.test)
sys.exit(not result.wasSuccessful())
main = TestProgram