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)