Make TestCodeHilite tests working with Pygments 2.0rc1.

Fixes Python-Markdown#361.
ibm-z-oss-oda · Nov 4, 2014 · f0357b2 · f0357b2
1 parent 0f008f7
commit f0357b2
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/tests/test_extensions.py b/tests/test_extensions.py
@@ -91,10 +91,8 @@ def testBasicCodeHilite(self):
         text = '\t# A Code Comment'
         md = markdown.Markdown(extensions=['markdown.extensions.codehilite'])
         if self.has_pygments:
-            self.assertEqual(md.convert(text),
-                '<div class="codehilite">'
-                '<pre><span class="c"># A Code Comment</span>\n'
-                '</pre></div>')
+            # Pygments can use random lexer here as we did not specify the language
+            self.assertTrue(md.convert(text).startswith('<div class="codehilite"><pre>'))
         else:
             self.assertEqual(md.convert(text),
                 '<pre class="codehilite"><code># A Code Comment'
@@ -105,7 +103,7 @@ def testLinenumsTrue(self):
         md = markdown.Markdown(
             extensions=[markdown.extensions.codehilite.CodeHiliteExtension(linenums=True)])
         if self.has_pygments:
-            # Differant versions of pygments output slightly different markup.
+            # Different versions of pygments output slightly different markup.
             # So we use 'startwith' and test just enough to confirm that 
             # pygments received and processed linenums.
             self.assertTrue(md.convert(text).startswith(
@@ -134,10 +132,8 @@ def testLinenumsNone(self):
         md = markdown.Markdown(
             extensions=[markdown.extensions.codehilite.CodeHiliteExtension(linenums=None)])
         if self.has_pygments:
-            self.assertEqual(md.convert(text),
-                '<div class="codehilite">'
-                '<pre><span class="c"># A Code Comment</span>\n'
-                '</pre></div>')
+            # Pygments can use random lexer here as we did not specify the language
+            self.assertTrue(md.convert(text).startswith('<div class="codehilite"><pre>'))
         else:
             self.assertEqual(md.convert(text),
                 '<pre class="codehilite"><code># A Code Comment'