diff options
Diffstat (limited to 'Lib/test/test_future.py')
-rw-r--r-- | Lib/test/test_future.py | 24 |
1 files changed, 11 insertions, 13 deletions
diff --git a/Lib/test/test_future.py b/Lib/test/test_future.py index c6689a1a18..a0c156f5f7 100644 --- a/Lib/test/test_future.py +++ b/Lib/test/test_future.py @@ -13,18 +13,18 @@ def get_error_location(msg): class FutureTest(unittest.TestCase): def test_future1(self): - support.unload('test_future1') - from test import test_future1 - self.assertEqual(test_future1.result, 6) + with support.CleanImport('future_test1'): + from test import future_test1 + self.assertEqual(future_test1.result, 6) def test_future2(self): - support.unload('test_future2') - from test import test_future2 - self.assertEqual(test_future2.result, 6) + with support.CleanImport('future_test2'): + from test import future_test2 + self.assertEqual(future_test2.result, 6) def test_future3(self): - support.unload('test_future3') - from test import test_future3 + with support.CleanImport('test_future3'): + from test import test_future3 def test_badfuture3(self): try: @@ -103,8 +103,8 @@ class FutureTest(unittest.TestCase): self.fail("syntax error didn't occur") def test_multiple_features(self): - support.unload("test.test_future5") - from test import test_future5 + with support.CleanImport("test.test_future5"): + from test import test_future5 def test_unicode_literals_exec(self): scope = {} @@ -112,8 +112,6 @@ class FutureTest(unittest.TestCase): self.assertIsInstance(scope["x"], str) -def test_main(): - support.run_unittest(FutureTest) if __name__ == "__main__": - test_main() + unittest.main() |