def test_highlight_python(self):
lang = 'python'
code = self.code[lang]
result = u'<div class="codehilite"><pre><span class="k">print</span> <span class="s">"Hello world!"</span>\n</pre></div>\n'
self.assertEquals(result, highlight(code, lang))
def test_highlight_plain_text(self):
code = self.code['plain']
self.assertEquals(code, highlight(code, 'plain'))
