mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
@@ -1,5 +1,7 @@
|
||||
import yaml
|
||||
import os
|
||||
from mock import MagicMock
|
||||
from contextlib import contextmanager
|
||||
|
||||
|
||||
def spy(method_to_decorate):
|
||||
@@ -16,3 +18,13 @@ def spy(method_to_decorate):
|
||||
def load_stage_file(path):
|
||||
with open(path, "r") as fobj:
|
||||
return yaml.safe_load(fobj)
|
||||
|
||||
|
||||
@contextmanager
|
||||
def cd(newdir):
|
||||
prevdir = os.getcwd()
|
||||
os.chdir(os.path.expanduser(newdir))
|
||||
try:
|
||||
yield
|
||||
finally:
|
||||
os.chdir(prevdir)
|
||||
|
||||
Reference in New Issue
Block a user