diff options
Diffstat (limited to 'Lib/unittest/test/test_break.py')
-rw-r--r-- | Lib/unittest/test/test_break.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/Lib/unittest/test/test_break.py b/Lib/unittest/test/test_break.py index 560001103f8..77ce20143bf 100644 --- a/Lib/unittest/test/test_break.py +++ b/Lib/unittest/test/test_break.py @@ -1,12 +1,10 @@ import gc +import io import os import sys import signal import weakref -from cStringIO import StringIO - - import unittest @@ -144,7 +142,7 @@ class TestBreak(unittest.TestCase): def testRunner(self): # Creating a TextTestRunner with the appropriate argument should # register the TextTestResult it creates - runner = unittest.TextTestRunner(stream=StringIO()) + runner = unittest.TextTestRunner(stream=io.StringIO()) result = runner.run(unittest.TestSuite()) self.assertIn(result, unittest.signals._results) @@ -211,7 +209,8 @@ class TestBreak(unittest.TestCase): self.assertEqual(FakeRunner.initArgs, [((), {'buffer': None, 'verbosity': verbosity, - 'failfast': failfast})]) + 'failfast': failfast, + 'warnings': None})]) self.assertEqual(FakeRunner.runArgs, [test]) self.assertEqual(p.result, result) @@ -224,7 +223,8 @@ class TestBreak(unittest.TestCase): self.assertEqual(FakeRunner.initArgs, [((), {'buffer': None, 'verbosity': verbosity, - 'failfast': failfast})]) + 'failfast': failfast, + 'warnings': None})]) self.assertEqual(FakeRunner.runArgs, [test]) self.assertEqual(p.result, result) |