lib/rouge.rb in rouge-0.2.3 vs lib/rouge.rb in rouge-0.2.4

@@ -5,9 +5,12 @@
   class << self
     def highlight(text, lexer, formatter)
       lexer = Lexer.find(lexer) unless lexer.respond_to? :lex
       raise "unknown lexer #{lexer}" unless lexer
 
+      formatter = Formatter.find(formatter) unless formatter.respond_to? :format
+      raise "unknown formatter #{formatter}" unless formatter
+
       formatter.format(lexer.lex(text))
     end
   end
 end