diff options
Diffstat (limited to 'Lib/test/test_socket.py')
-rw-r--r-- | Lib/test/test_socket.py | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py index 279fd90..b9650e7 100644 --- a/Lib/test/test_socket.py +++ b/Lib/test/test_socket.py @@ -683,17 +683,18 @@ class SmallBufferedFileObjectClassTestCase(FileObjectClassTestCase): bufsize = 2 # Exercise the buffering code def test_main(): - suite = unittest.TestSuite() - suite.addTest(unittest.makeSuite(GeneralModuleTests)) - suite.addTest(unittest.makeSuite(BasicTCPTest)) + tests = [ GeneralModuleTests, BasicTCPTest ] if sys.platform != 'mac': - suite.addTest(unittest.makeSuite(BasicUDPTest)) - suite.addTest(unittest.makeSuite(NonBlockingTCPTests)) - suite.addTest(unittest.makeSuite(FileObjectClassTestCase)) - suite.addTest(unittest.makeSuite(UnbufferedFileObjectClassTestCase)) - suite.addTest(unittest.makeSuite(LineBufferedFileObjectClassTestCase)) - suite.addTest(unittest.makeSuite(SmallBufferedFileObjectClassTestCase)) - test_support.run_suite(suite) + tests.append(BasicUDPTest) + + tests.extend([ + NonBlockingTCPTests, + FileObjectClassTestCase, + UnbufferedFileObjectClassTestCase, + LineBufferedFileObjectClassTestCase, + SmallBufferedFileObjectClassTestCase + ]) + test_support.run_unittest(*tests) if __name__ == "__main__": test_main() |