Add colorize

test/test_lexer.rb

@@ -34,4 +34,20 @@ class TestLexer < Test::Unit::TestCase
     assert !Lexer['Ruby'].eql?(Lexer['Python'])
     assert !Lexer['Ruby'].eql?(Lexer.new('Ruby'))
   end
+
+  def test_colorize
+    assert_equal <<-HTML, Lexer['Text only'].colorize("Hello")
+<div class="highlight"><pre>Hello
+</pre>
+</div>
+    HTML
+
+    assert_equal <<-HTML, Lexer['Ruby'].colorize("def foo\n  'foo'\nend\n")
+<div class="highlight"><pre><span class="k">def</span> <span class="nf">foo</span>
+  <span class="s1">&#39;foo&#39;</span>
+<span class="k">end</span>
+</pre>
+</div>
+    HTML
+  end
 end