reclassifier 0.4.9 → 0.4.10

data/lib/reclassifier/bayes.rb

@@ -105,7 +105,7 @@ class Reclassifier::Bayes
   #  =>  :uninteresting
   #
   def classify(text)
-    calculate_scores(text).max_by {|classification| classification[1]}[0]
+    calculate_scores(text.to_s).max_by {|classification| classification[1]}[0]
   end
 
   # Provides a list of classification names

data/lib/reclassifier/version.rb

@@ -1,3 +1,3 @@
 module Reclassifier
-  VERSION = "0.4.9"
+  VERSION = "0.4.10"
 end

data/spec/bayes_spec.rb

@@ -13,6 +13,17 @@ describe Reclassifier::Bayes do
     end
   end
 
+  describe 'classify' do
+    it 'should handle a nil text gracefully' do
+      classifier = Reclassifier::Bayes.new([:in_china, :not_in_china])
+
+      classifier.train(:in_china, 'Chinese Macao')
+      classifier.train(:not_in_china, 'Tokyo Japan Chinese')
+
+      classifier.classify(nil)
+    end
+  end
+
 	describe "classifications" do
     it "should return the classifications" do
       classifier = Reclassifier::Bayes.new([:interesting, :uninteresting])

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: reclassifier
 version: !ruby/object:Gem::Version
-  version: 0.4.9
+  version: 0.4.10
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-04-25 00:00:00.000000000 Z
+date: 2013-06-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler