diff Lib/importlib/test/source/test_file_loader.py @ 55737:b65ebed06eb0

Move over to using assertRaises as a context manager for importlib tests. Obviously one shouldn't do whole sale conversions like this, but I was already going through the test code and I was bored at the airport.
author Brett Cannon <bcannon@gmail.com>
date Thu, 27 Aug 2009 23:49:21 +0000
parents 0bb232078b8c
children 27e120f9208b
line wrap: on
line diff
--- a/Lib/importlib/test/source/test_file_loader.py	Thu Aug 27 23:46:38 2009 +0000
+++ b/Lib/importlib/test/source/test_file_loader.py	Thu Aug 27 23:49:21 2009 +0000
@@ -93,7 +93,8 @@
                 file.write('+++ bad syntax +++')
             loader = _bootstrap._PyPycFileLoader('_temp', mapping['_temp'],
                                                     False)
-            self.assertRaises(SyntaxError, loader.load_module, name)
+            with self.assertRaises(SyntaxError):
+                loader.load_module(name)
             for attr in attributes:
                 self.assertEqual(getattr(orig_module, attr), value)
 
@@ -104,7 +105,8 @@
                 file.write('=')
             loader = _bootstrap._PyPycFileLoader('_temp', mapping['_temp'],
                                                     False)
-            self.assertRaises(SyntaxError, loader.load_module, '_temp')
+            with self.assertRaises(SyntaxError):
+                loader.load_module('_temp')
             self.assertTrue('_temp' not in sys.modules)
 
 
@@ -166,8 +168,8 @@
                 bytecode_file.write(imp.get_magic())
                 bytecode_file.write(source_timestamp)
                 bytecode_file.write(b'AAAA')
-            self.assertRaises(ValueError, self.import_, mapping['_temp'],
-                                '_temp')
+            with self.assertRaises(ValueError):
+                self.import_(mapping['_temp'], '_temp')
             self.assertTrue('_temp' not in sys.modules)