move environment to helpers module

This fixes registering pytest plugins
This commit is contained in:
Jörg Thalheim
2023-07-28 12:10:07 +02:00
committed by Mic92
parent 282307ca2a
commit df6e94b20e
5 changed files with 5 additions and 2 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)