Guard against classify nil data

2 files changed, 5 insertions, 0 deletions

diff --git a/lib/linguist/classifier.rb b/lib/linguist/classifier.rb
index dd3fe04..30fb7c4 100644
--- a/lib/linguist/classifier.rb
+++ b/lib/linguist/classifier.rb
@@ -97,6 +97,7 @@ module Linguist
     # Returns sorted Array of result pairs. Each pair contains the
     # Language and a Float score.
     def classify(tokens, languages = @languages.keys)
+      return [] if tokens.nil?
       tokens = Tokenizer.new(tokens).tokens if tokens.is_a?(String)
 
       scores = {}
diff --git a/test/test_classifier.rb b/test/test_classifier.rb
index b39fb20..508aa08 100644
--- a/test/test_classifier.rb
+++ b/test/test_classifier.rb
@@ -55,6 +55,10 @@ class TestClassifier < Test::Unit::TestCase
     assert results.first[1] < 0.5, results.first.inspect
   end
 
+  def test_instance_classify_nil
+    assert_equal [], Classifier.instance.classify(nil)
+  end
+
   def test_verify
     assert Classifier.instance.verify
   end