1
0
forked from VimPlug/jedi

Start using our own monkeypatch function for some things

This commit is contained in:
Dave Halter
2018-04-20 21:34:00 +02:00
parent ceb5509170
commit 0bcd1701f0
2 changed files with 16 additions and 5 deletions
+14
View File
@@ -1,4 +1,5 @@
import os
from contextlib import contextmanager
def traverse_parents(path, include_current=False):
@@ -10,3 +11,16 @@ def traverse_parents(path, include_current=False):
yield path
previous = path
path = os.path.dirname(path)
@contextmanager
def monkeypatch(obj, attribute_name, new_value):
"""
Like pytest's monkeypatch, but as a context manager.
"""
old_value = getattr(obj, attribute_name)
try:
setattr(obj, attribute_name, new_value)
yield
finally:
setattr(obj, attribute_name, old_value)