summaryrefslogtreecommitdiff
path: root/Lib/test/test_csv.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_csv.py')
-rw-r--r--Lib/test/test_csv.py18
1 files changed, 7 insertions, 11 deletions
diff --git a/Lib/test/test_csv.py b/Lib/test/test_csv.py
index c4d1a03bb1..18f8851e70 100644
--- a/Lib/test/test_csv.py
+++ b/Lib/test/test_csv.py
@@ -6,7 +6,7 @@ import unittest
from StringIO import StringIO
import csv
import gc
-from test.test_support import verbose
+from test import test_support
class Test_Csv(unittest.TestCase):
"""
@@ -568,7 +568,7 @@ Stonecutters Seafood and Chop House, Lemont, IL, 12/19/02, Week Back
self.assertEqual(dialect.skipinitialspace, False)
if not hasattr(sys, "gettotalrefcount"):
- if verbose: print "*** skipping leakage tests ***"
+ if test_support.verbose: print "*** skipping leakage tests ***"
else:
class NUL:
def write(s, *args):
@@ -640,15 +640,11 @@ else:
# if writer leaks during write, last delta should be 5 or more
self.assertEqual(delta < 5, True)
-def _testclasses():
+def test_main():
mod = sys.modules[__name__]
- return [getattr(mod, name) for name in dir(mod) if name.startswith('Test')]
-
-def suite():
- suite = unittest.TestSuite()
- for testclass in _testclasses():
- suite.addTest(unittest.makeSuite(testclass))
- return suite
+ test_support.run_unittest(
+ *[getattr(mod, name) for name in dir(mod) if name.startswith('Test')]
+ )
if __name__ == '__main__':
- unittest.main(defaultTest='suite')
+ test_main()