Function temp_env

tests/operators/test_tree_mask.py:39–48

(key, value)

@contextmanager
def temp_env(key, value):
    old_val = os.environ.get(key)
    os.environ[key] = value
    try:
        yield
    finally:
        if old_val is None:
            os.environ.pop(key, None)
        else:
            os.environ[key] = old_val


class TestTreeMask(unittest.TestCase):