From 372d3e8c78512b26bfb44ec1ef505ff772bd9302 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Wed, 5 Mar 2025 10:31:11 +1000 Subject: [PATCH] tidy: remove unused utils --- tests/env_var_utils.py | 25 ------------------------- 1 file changed, 25 deletions(-) delete mode 100644 tests/env_var_utils.py diff --git a/tests/env_var_utils.py b/tests/env_var_utils.py deleted file mode 100644 index 5bf0c1d3490..00000000000 --- a/tests/env_var_utils.py +++ /dev/null @@ -1,25 +0,0 @@ -import os -from contextlib import contextmanager - - -@contextmanager -def unset_env_var(env_var: str): - """Context manager that unsets an environment variable, re-setting it to the original value when the context is exited.""" - prev_value = os.environ.get(env_var, None) - if prev_value is not None: - del os.environ[env_var] - yield - if prev_value is not None: - os.environ[env_var] = prev_value - - -@contextmanager -def set_env_var(env_var: str, value: str): - """Context manager that sets an environment variable, re-setting it to the original value when the context is exited.""" - prev_value = os.environ.get(env_var, None) - os.environ[env_var] = value - yield - if prev_value is not None: - os.environ[env_var] = prev_value - else: - del os.environ[env_var]