From: Raymond Hettinger <python@rcn.com>
Date: Fri, 12 Jun 2009 11:25:59 +0000 (+0000)
Subject: Fixup/simplify another nested context manager.
X-Git-Tag: v3.1rc2~7
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=554290d9209f0a552093f03e3c8c8e512b2a0efa;p=python

Fixup/simplify another nested context manager.
---

diff --git a/Lib/importlib/test/import_/test_meta_path.py b/Lib/importlib/test/import_/test_meta_path.py
index 59f11d3836..69859b1bbf 100644
--- a/Lib/importlib/test/import_/test_meta_path.py
+++ b/Lib/importlib/test/import_/test_meta_path.py
@@ -1,6 +1,5 @@
 from .. import util
 from . import util as import_util
-from contextlib import nested
 from types import MethodType
 import unittest
 
@@ -18,8 +17,7 @@ class CallingOrder(unittest.TestCase):
         mod = 'top_level'
         first = util.mock_modules(mod)
         second = util.mock_modules(mod)
-        context = nested(util.mock_modules(mod), util.mock_modules(mod))
-        with  context as (first, second):
+        with util.mock_modules(mod) as first, util.mock_modules(mod) as second:
             first.modules[mod] = 42
             second.modules[mod] = -13
             with util.import_state(meta_path=[first, second]):
@@ -28,9 +26,8 @@ class CallingOrder(unittest.TestCase):
     def test_continuing(self):
         # [continuing]
         mod_name = 'for_real'
-        first = util.mock_modules('nonexistent')
-        second = util.mock_modules(mod_name)
-        with nested(first, second):
+        with util.mock_modules('nonexistent') as first, \
+             util.mock_modules(mod_name) as second:
             first.find_module = lambda self, fullname, path=None: None
             second.modules[mod_name] = 42
             with util.import_state(meta_path=[first, second]):