Simplify clean_env() fixture

Simplify the clean_env() fixture to build upon pytest's monkeypatch fixture.
click-contrib · Nov 24, 2023 · 5c50df4 · 5c50df4
1 parent 354ed73
commit 5c50df4
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,5 +1,3 @@
-import os
-
 import pytest
 
 import click
@@ -12,9 +10,6 @@ def runner():
 
 
 @pytest.fixture(autouse=True)
-def clean_env():
-    old = dict(os.environ)
-    os.environ.pop("FORCE_COLOR", None)
-    os.environ.pop("NO_COLOR", None)
-    yield
-    os.environ = old
+def clean_env(monkeypatch):
+    monkeypatch.delenv("FORCE_COLOR", raising=False)
+    monkeypatch.delenv("NO_COLOR", raising=False)