Merge pull request #1 from theverything/get_score

Refactor #get_score

lib/sentimental.rb

@@ -64,18 +64,10 @@ def get_score(string)
     #tokenize the string, also throw away some punctuation
     tokens = string.to_s.downcase.split(/[\s\!\?\.]+/)
 
-    for token in tokens do
+    tokens.each do |token|
       sentiment_value = @@sentihash[token]
-
-      if sentiment_value
-
-        # for debugging purposes
-        #puts "#{token} => #{sentiment_value}"
-
-        sentiment_total += sentiment_value
-      end
+      sentiment_total += sentiment_value if sentiment_value
     end
-
     sentiment_total
   end
 

spec/sentimental_spec.rb

@@ -0,0 +1,27 @@
+require_relative "../lib/sentimental"
+
+describe Sentimental do
+
+   before :each do
+    Sentimental.load_defaults
+    Sentimental.threshold = 0.1
+    @analyzer = Sentimental.new
+  end
+
+  describe "#get_score" do
+
+    it "returns a score of 0.925 when passed 'I love ruby'" do
+      @analyzer.get_score('I love ruby').should eq(0.925)
+    end
+
+    it "returns a score of 0.0 when passed 'I like ruby'" do
+      @analyzer.get_score('I like ruby').should eq(0.0)
+    end
+
+    it "returns a score of -0.4375 when passed 'I hate ruby'" do
+      @analyzer.get_score('I hate ruby').should eq(-0.4375)
+    end
+
+  end
+
+end