def test_env_var_not_already_set():
""" Tests setting and env var when it's not already set """
env_var = "model-zoo-test-env-var-name"
new_value = "new_value"
try:
# Make sure that the env var is unset to start
if os.environ.get(env_var):
del os.environ[env_var]
# Try setting the value with the overwrite flag set to False
set_env_var(env_var, new_value, overwrite_existing=False)
# Verify that we now have a value
assert os.environ[env_var] == new_value
# Unset the env var and set it with the overwrite flag set to True
del os.environ[env_var]
new_value = "another_new_value"
set_env_var(env_var, new_value, overwrite_existing=True)
# Verify that we have the new value set
assert os.environ[env_var] == new_value
finally:
if os.environ.get(env_var):
del os.environ[env_var]
def test_env_var_already_set():
""" Tests setting and env var when it's already set """
env_var = "model-zoo-test-env-var-name"
original_value = "original"
modified_value = "modified"
try:
# Set the env var to an initial value
os.environ[env_var] = original_value
# Try to modify that value but set overwrite flag to False
set_env_var(env_var, modified_value, overwrite_existing=False)
# Verify that we still have the original value
assert os.environ[env_var] == original_value
# Try to modify the value with the overwrite flag set to True
set_env_var(env_var, modified_value, overwrite_existing=True)
# Verify that we now have the modified value
assert os.environ[env_var] == modified_value
finally:
if os.environ.get(env_var):
del os.environ[env_var]