diff options
Diffstat (limited to 'Lib/importlib/test/frozen')
| -rw-r--r-- | Lib/importlib/test/frozen/test_loader.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/Lib/importlib/test/frozen/test_loader.py b/Lib/importlib/test/frozen/test_loader.py index fa64f30..c05e22c 100644 --- a/Lib/importlib/test/frozen/test_loader.py +++ b/Lib/importlib/test/frozen/test_loader.py @@ -51,8 +51,8 @@ class LoaderTests(abc.LoaderTests): def test_unloadable(self): assert machinery.FrozenImporter.find_module('_not_real') is None - self.assertRaises(ImportError, machinery.FrozenImporter.load_module, - '_not_real') + with self.assertRaises(ImportError): + machinery.FrozenImporter.load_module('_not_real') class InspectLoaderTests(unittest.TestCase): @@ -84,7 +84,8 @@ class InspectLoaderTests(unittest.TestCase): # Raise ImportError for modules that are not frozen. for meth_name in ('get_code', 'get_source', 'is_package'): method = getattr(machinery.FrozenImporter, meth_name) - self.assertRaises(ImportError, method, 'importlib') + with self.assertRaises(ImportError): + method('importlib') def test_main(): |