diff options
| author | Nick Coghlan <ncoghlan@gmail.com> | 2011-05-05 13:49:25 (GMT) |
|---|---|---|
| committer | Nick Coghlan <ncoghlan@gmail.com> | 2011-05-05 13:49:25 (GMT) |
| commit | 0ded3e307b188246018c893d90f26dfba6fe282c (patch) | |
| tree | e38e7fb0d4d1a7de987713b7ea540dfdf00ea230 /Lib/test/test_with.py | |
| parent | f77b74dd1beacb69b6853303aa70a0de64cd71ae (diff) | |
| download | cpython-0ded3e307b188246018c893d90f26dfba6fe282c.zip cpython-0ded3e307b188246018c893d90f26dfba6fe282c.tar.gz cpython-0ded3e307b188246018c893d90f26dfba6fe282c.tar.bz2 | |
Issue #11647: allow contextmanager objects to be used as decorators as described in the docs. Initial patch by Ysj Ray.
Diffstat (limited to 'Lib/test/test_with.py')
| -rw-r--r-- | Lib/test/test_with.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/Lib/test/test_with.py b/Lib/test/test_with.py index a9d374b..e8cc8c0 100644 --- a/Lib/test/test_with.py +++ b/Lib/test/test_with.py @@ -14,8 +14,8 @@ from test.support import run_unittest class MockContextManager(_GeneratorContextManager): - def __init__(self, gen): - _GeneratorContextManager.__init__(self, gen) + def __init__(self, func, *args, **kwds): + super().__init__(func, *args, **kwds) self.enter_called = False self.exit_called = False self.exit_args = None @@ -33,7 +33,7 @@ class MockContextManager(_GeneratorContextManager): def mock_contextmanager(func): def helper(*args, **kwds): - return MockContextManager(func(*args, **kwds)) + return MockContextManager(func, *args, **kwds) return helper |