KevinMidboe/linguist
1
0
Fork 0
mirror of https://github.com/KevinMidboe/linguist.git synced 2025-10-29 17:50:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Guard against classify nil data

Browse Source
This commit is contained in:
Joshua Peek
2012-06-21 11:47:32 -05:00
parent 3d7364877d
commit 2b712dc790
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lib/linguist/classifier.rb
View File

@@ -97,6 +97,7 @@ module Linguist
# Returns sorted Array of result pairs. Each pair contains the # Returns sorted Array of result pairs. Each pair contains the
# Language and a Float score. # Language and a Float score.
def classify(tokens, languages = @languages.keys) def classify(tokens, languages = @languages.keys)
return [] if tokens.nil?
tokens = Tokenizer.new(tokens).tokens if tokens.is_a?(String) tokens = Tokenizer.new(tokens).tokens if tokens.is_a?(String)
scores = {} scores = {}

4
test/test_classifier.rb
View File

@@ -55,6 +55,10 @@ class TestClassifier < Test::Unit::TestCase
assert results.first[1] < 0.5, results.first.inspect assert results.first[1] < 0.5, results.first.inspect
end end
def test_instance_classify_nil
assert_equal [], Classifier.instance.classify(nil)
end
def test_verify def test_verify
assert Classifier.instance.verify assert Classifier.instance.verify
end end