OpenMC/tests/__init__.py

16 lines
336 B
Python
Raw Permalink Normal View History

2019-10-28 11:55:45 -05:00
from contextlib import contextmanager
import os
import tempfile
@contextmanager
def cdtemp():
"""Context manager to change to/return from a tmpdir."""
with tempfile.TemporaryDirectory() as tmpdir:
cwd = os.getcwd()
try:
os.chdir(tmpdir)
yield
finally:
os.chdir(cwd)