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]