15 lines
320 B
Python
15 lines
320 B
Python
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)
|