move moc_env to module

This commit is contained in:
Jörg Thalheim
2023-07-26 15:14:29 +02:00
parent a61d0c5a42
commit b4ba9c70cd
3 changed files with 17 additions and 14 deletions

View File

@@ -0,0 +1,14 @@
import os
from contextlib import contextmanager
from typing import Iterator
@contextmanager
def mock_env(**environ: str) -> Iterator[None]:
original_environ = dict(os.environ)
os.environ.update(environ)
try:
yield
finally:
os.environ.clear()
os.environ.update(original_environ)