environ_decorator.py

from functools import wraps
from mock import patch


def environ(**variables):
    def decorator(func):
        @wraps(func)
        def wrapped(*args, **kwargs):
            with patch.dict('os.environ', variables):
                return func(*args, **kwargs)
        return wrapped
    return decorator