Merge pull request #26 from mgorny/nocolor-pytest-r1

Simplify clean_env() fixture
click-contrib · Nov 25, 2023 · 5d42a55 · 5d42a55
2 parents 354ed73 + 5c50df4
commit 5d42a55
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)