Fix code highlighting on linux

Erotemic · Sep 11, 2020 · 15cb51b · 15cb51b
1 parent a6a7b7f
commit 15cb51b
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/xdoctest/utils/util_str.py b/xdoctest/utils/util_str.py
@@ -194,10 +194,14 @@ def highlight_code(text, lexer_name='python', **kwargs):
         'cxx': 'cpp',
         'c': 'cpp',
     }.get(lexer_name.replace('.', ''), lexer_name)
-    import os
-    if os.environ.get('XDOC_WIN32_COLORS', 'False') == 'False':
-        # hack: dont color on windows by default, but do init colorama
-        return text
+
+    if sys.platform.startswith('win32'):  # nocover
+        # Hack on win32 to support colored output
+        import os
+        if os.environ.get('XDOC_WIN32_COLORS', 'False') == 'False':
+            # hack: dont color on windows by default, but do init colorama
+            return text
+
     try:
         import pygments
         import pygments.lexers