Преглед изворни кода

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)