ソースを参照

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:
         def restore() -> None:
             if oldval is not None:
             if oldval is not None:
                 os.environ[name] = oldval
                 os.environ[name] = oldval
-            else:
+            elif name in os.environ:
                 del os.environ[name]
                 del os.environ[name]
 
 
         oldval = os.environ.get(name)
         oldval = os.environ.get(name)
         if value is not None:
         if value is not None:
             os.environ[name] = value
             os.environ[name] = value
-        else:
+        elif name in os.environ:
             del os.environ[name]
             del os.environ[name]
         self.addCleanup(restore)
         self.addCleanup(restore)