davidrichards-just_enumerable_stats 0.0.13 → 0.0.14

data/VERSION.yml

@@ -1,4 +1,4 @@
 --- 
 :major: 0
 :minor: 0
-:patch: 13
+:patch: 14

data/lib/just_enumerable_stats.rb

@@ -1,4 +1,5 @@
 require 'rubygems'
+require 'mathn'
 
 # Need the FixedRange
 $:.unshift File.dirname(__FILE__)
@@ -765,7 +766,22 @@ module Enumerable
     self._jes_scale!{|e| (e * scalar) + shift}
   end
   safe_alias :_jes_scale_between!
+  
+  # Returns a hash or dictionary (if installed) of the frequency of each category.
+  def _jes_frequency
+    dict = defined?(Dictionary) ? Dictionary.new : Hash.new
+    self._jes_category_values.each do |k, v|
+      dict[k] = v.size / self.size
+    end
+    dict
+  end
+  safe_alias :_jes_frequency
+  
+  def _jes_frequency_for(key)
+    self._jes_frequency[key]
+  end
+  safe_alias :_jes_frequency_for
 
 end
 
-@a = [1,2,3]
+# @a = [1,2,3]

data/spec/just_enumerable_stats_spec.rb

@@ -162,19 +162,19 @@ describe "JustEnumerableStats" do
   end
   
   it "should be able to calculate the standard deviation" do
-    @a.standard_deviation.should eql(1.0)
-    @a.std.should eql(1.0)
+    @a.standard_deviation.should eql(1)
+    @a.std.should eql(1)
   end
   
   it "should be able to calculate the standard deviation with a block" do
-    @a.standard_deviation(&@doubler).should eql(2.0)
-    @a.std(&@doubler).should eql(2.0)
+    @a.standard_deviation(&@doubler).should eql(2)
+    @a.std(&@doubler).should eql(2)
   end
   
   it "should be able to calculate the standard deviation with a default block" do
     @a.default_block = @doubler
-    @a.standard_deviation.should eql(2.0)
-    @a.std.should eql(2.0)
+    @a.standard_deviation.should eql(2)
+    @a.std.should eql(2)
   end
   
   it "should be able to calculate the median value" do
@@ -316,6 +316,10 @@ describe "JustEnumerableStats" do
     @b.rank.should eql([1,2,3])
   end
   
+  it "should have ordinalize work like ranke" do
+    @b.ordinalize.should eql([3,2,1])
+  end
+  
   it "should be able to get the order of values, handling duplicates" do
     [10,5,5,1].order.should eql([4,2,3,1])
   end
@@ -601,4 +605,17 @@ describe "JustEnumerableStats" do
       @a.should eql(b)
     end
   end
+  
+  context "frequency" do
+    it "should be able to generate a frequency" do
+      @a.frequency[1].should eql(1/3)
+      @a.frequency[2].should eql(1/3)
+      @a.frequency[3].should eql(1/3)
+    end
+    
+    it "should be able to generate a frequency for a specific value" do
+      a = [3,3,3,2]
+      a.frequency_for(3).should eql(3/4)
+    end
+  end
 end

data/spec/spec_helper.rb

@@ -72,6 +72,8 @@ class BusyClass
   def normalize!; raise ArgumentError, "Should not be called"; end
   def scale_between; raise ArgumentError, "Should not be called"; end
   def scale_between!; raise ArgumentError, "Should not be called"; end
+  def frequency; raise ArgumentError, "Should not be called"; end
+  def frequency_for(val); raise ArgumentError, "Should not be called"; end
 end
 
     

data/spec/unobtrusive_just_enumerable_stats_spec.rb

@@ -227,6 +227,14 @@ describe "JustEnumerableStats" do
     it "should not use the native scale_between!" do
       lambda{@a._jes_scale_between!(6,8)}.should_not raise_error
     end
+
+    it "should not use the native frequency" do
+      lambda{@a._jes_frequency}.should_not raise_error
+    end
+    
+    it "should not use the native frequency_for" do
+      lambda{@a._jes_frequency_for(2)}.should_not raise_error
+    end
     
   end
   

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: davidrichards-just_enumerable_stats
 version: !ruby/object:Gem::Version 
-  version: 0.0.13
+  version: 0.0.14
 platform: ruby
 authors: 
 - David Richards