Decision - extract method record_count

asafschers · Mar 3, 2018 · 4c6d855 · 4c6d855
1 parent e5bc0b6
commit 4c6d855
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/scoruby/decision.rb b/lib/scoruby/decision.rb
@@ -21,7 +21,11 @@ def initialize(xml)
     def probability(score_distribution, xml)
       probability = score_distribution.attributes['probability'].to_s
       return probability.to_f if probability != ''
-      score_distribution.attributes['recordCount'].to_s.to_f / xml.attributes['recordCount'].to_s.to_f
+      record_count(score_distribution) / record_count(xml)
+    end
+
+    def record_count(xml)
+      xml.attributes['recordCount'].to_s.to_f
     end
   end
 end