Improve environment variable fetching in tests

Uses the os.environ.get method instead of a dict-like lookup and a try-except block