瀏覽代碼

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)