OpenMC/tests/__init__.py
2018-02-08 14:32:28 -06:00

15 lines
336 B
Python

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)