Skip pygments-related tests if it is not installed.

1 files changed, 11 insertions, 4 deletions

diff --git a/tests/test_highlighting.py b/tests/test_highlighting.py
index 198f7a0a..ea1f25f1 100644
--- a/tests/test_highlighting.py
+++ b/tests/test_highlighting.py
@@ -11,6 +11,12 @@
 
 from util import *
 
+try:
+    import pygments
+except ImportError:
+    from nose.plugins.skip import SkipTest
+    raise SkipTest('pygments not available')
+
 from pygments.lexer import RegexLexer
 from pygments.token import Text, Name
 from pygments.formatters.html import HtmlFormatter
@@ -28,16 +34,17 @@ class MyLexer(RegexLexer):
         ],
     }
 
-class ComplainOnUnhighlighted(PygmentsBridge):
-
-    def unhighlighted(self, source):
-        raise AssertionError("should highlight %r" % source)
 
 class MyFormatter(HtmlFormatter):
     def format(self, tokensource, outfile):
         outfile.write('test')
 
 
+class ComplainOnUnhighlighted(PygmentsBridge):
+    def unhighlighted(self, source):
+        raise AssertionError("should highlight %r" % source)
+
+
 @with_app()
 def test_add_lexer(app):
     app.add_lexer('test', MyLexer())