Merge pull request #25 from mgorny/nocolor-pytest

Fix running the test suite when NO_COLOR is set in calling env
click-contrib · Nov 24, 2023 · 354ed73 · 354ed73
2 parents 6042d4c + 7603764
commit 354ed73
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,3 +1,5 @@
+import os
+
 import pytest
 
 import click
@@ -7,3 +9,12 @@
 @pytest.fixture
 def runner():
  return CliRunner()
+
+
+@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