import os
from contextlib import contextmanager
from typing import Union, ContextManager

PathType = Union[os.PathLike, str]


@contextmanager
def cd(path: PathType) -> ContextManager[None]:
    old_path = Path.cwd()
    os.chdir(path)
    yield
    os.chdir(old_path)