Explorar o código

Handle env variables no longer being set

Jelmer Vernooij hai 3 semanas
pai
achega
01f0c2bc4f
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  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)