Make BetterLoader() have a better api :)

This commit is contained in:
Armin Ronacher 2011-08-26 12:09:55 +01:00
parent a082a5e0ba
commit 5235c3e37e
2 changed files with 5 additions and 3 deletions

View file

@ -157,8 +157,7 @@ class BetterLoader(unittest.TestLoader):
all_tests.append(testcase)
if not all_tests:
print >> sys.stderr, 'Error: could not find test case for "%s"' % name
sys.exit(1)
raise LookupError('could not find test case for "%s"' % name)
if len(all_tests) == 1:
return all_tests[0]

View file

@ -1,3 +1,6 @@
import unittest
from flask.testsuite import BetterLoader
unittest.main(testLoader=BetterLoader(), defaultTest='suite')
try:
unittest.main(testLoader=BetterLoader(), defaultTest='suite')
except Exception, e:
print 'Error: %s' % e