tests: Correct way to remove accidental env vars

locustio · Jan 5, 2025 · 5a6def7 · 5a6def7
1 parent fcb6104
commit 5a6def7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/locust/test/__init__.py b/locust/test/__init__.py
@@ -17,6 +17,6 @@
 import os
 
 # Clean environment from stuff you might have lying around
-del os.environ["LOCUST_WEB_LOGIN"]
+os.environ.pop("LOCUST_WEB_LOGIN", None)
 # Ensure locust-cloud is not loaded during tests
-del os.environ["PGHOST"]
+os.environ.pop("PGHOST", None)