소스 검색

Handle env variables no longer being set

Jelmer Vernooij 3 주 전
부모
커밋
01f0c2bc4f
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      tests/__init__.py

+ 2 - 2
tests/__init__.py

@@ -53,13 +53,13 @@ class TestCase(_TestCase):
         def restore() -> None:
             if oldval is not None:
                 os.environ[name] = oldval
-            else:
+            elif name in os.environ:
                 del os.environ[name]
 
         oldval = os.environ.get(name)
         if value is not None:
             os.environ[name] = value
-        else:
+        elif name in os.environ:
             del os.environ[name]
         self.addCleanup(restore)