davidrichards-just_enumerable_stats 0.0.14 → 0.0.15

data/VERSION.yml

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

data/lib/just_enumerable_stats.rb

@@ -294,9 +294,49 @@ module Enumerable
       @_jes_categories = hash.keys
       @_jes_range_hash = hash
     end
+    @_jes_category_values = nil
     @_jes_categories
   end
   safe_alias :_jes_set_range
+  safe_alias :_jes_set_categories, :_jes_set_range
+  safe_alias :set_categories, :_jes_set_range
+  
+  # Allows you to add one category at a time.  You can actually add more
+  # than one category at a time, but it won't disrupt any previously-set
+  # categories. 
+  def _jes_add_category(hash)
+    _jes_init_range_hash
+    hash.each do |k, v|
+      @_jes_range_hash[k] = v
+    end
+    @_jes_category_values = nil
+    hash
+  end
+  safe_alias :_jes_add_category
+  
+  def method_missing(sym, *args, &block)
+    if self.categories.include?(sym)
+      self._jes_render_category(sym)
+    else
+      super
+    end
+  end
+
+  # Returns a specific category's values
+  def _jes_render_category(category)
+    self.category_values[category]
+  end
+  
+  def _jes_init_range_hash
+    if defined?(Dictionary)
+      @_jes_range_hash ||= Dictionary.new
+      @_jes_categories ||= []
+    else
+      @_jes_range_hash ||= {}
+      @_jes_categories ||= []
+    end
+  end
+  protected :_jes_init_range_hash
   
   # The hash of lambdas that are used to categorize the enumerable.
   attr_reader :_jes_range_hash

data/spec/just_enumerable_stats_spec.rb

@@ -240,6 +240,26 @@ describe "JustEnumerableStats" do
     })
     @a.categories.sort.should eql(["<= 2", "> 2"].sort)
   end
+
+  it "should be able to set a category as an alias to set_range" do
+    @a.set_categories({
+      "<= 2" => lambda{ |e| e <= 2 },
+      "> 2" => lambda{ |e| e > 2 }
+    })
+    @a.categories.sort.should eql(["<= 2", "> 2"].sort)
+  end
+  
+  it "should be able to add a category" do
+    @a.set_categories({
+      "<= 2" => lambda{ |e| e <= 2 },
+      "== 2" => lambda{ |e| e == 2 }
+    })
+    @a.category_values["<= 2"].should eql([1,2])
+    @a.category_values["== 2"].should eql([2])
+    @a.add_category({">= 2" => lambda{ |e| e >= 2}})
+    @a.categories.sort.should eql([">= 2", "<= 2", "== 2"].sort)
+    @a.category_values[">= 2"].should eql([2,3])
+  end
   
   it "should be able to get a hash of category values" do
     @a.set_range({
@@ -250,6 +270,14 @@ describe "JustEnumerableStats" do
     @a.category_values["> 2"].should eql([3])
   end
   
+  it "should be able to get categories with dot notation" do
+    @a.set_range({
+      :small => lambda{ |e| e <= 2 },
+      :large => lambda{ |e| e > 2 }
+    })
+    @a.small.should eql([1,2])
+  end
+  
   it "should be able to get category values with a regular range" do
     @a.category_values[1].should eql([1])
     @a.category_values[2].should eql([2])

data/spec/spec_helper.rb

@@ -7,6 +7,13 @@ Spec::Runner.configure do |config|
   
 end
 
+
+# This is one monster class that happens to have everything that we want
+# to define with this gem.  Since all the definitions raise an error,
+# all we have to do is call the _jes_ version of each method and make
+# sure that no error is raised.  This helps us make sure there are no
+# dependencies from within the method that use a non-jes version of a
+# method. 
 class BusyClass
   include Enumerable
   def initialize(*vals)
@@ -33,6 +40,8 @@ class BusyClass
   def range(&block); raise ArgumentError, "Should not be called"; end
   def set_range_class(klass, *args); raise ArgumentError, "Should not be called"; end
   def set_range(hash); raise ArgumentError, "Should not be called"; end
+  def set_categories(hash); raise ArgumentError, "Should not be called"; end
+  def add_category(hash); raise ArgumentError, "Should not be called"; end
   def dichotomize(split_value, first_label, second_label); raise ArgumentError, "Should not be called"; end
   def count_if(&block); raise ArgumentError, "Should not be called"; end
   def category_values(reset=false); raise ArgumentError, "Should not be called"; end

data/spec/unobtrusive_just_enumerable_stats_spec.rb

@@ -72,6 +72,14 @@ describe "JustEnumerableStats" do
       lambda{@a._jes_set_range({:a => 1})}.should_not raise_error
     end
     
+    it "should not use the native set_categories" do
+      lambda{@a._jes_set_categories({:a => 1})}.should_not raise_error
+    end
+    
+    it "should not use the native add_category" do
+      lambda{@a._jes_add_category({:a => 1})}.should_not raise_error
+    end
+    
     it "should not use the native dichotomize" do
       lambda{@a._jes_dichotomize(2, :small, :big)}.should_not raise_error
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: davidrichards-just_enumerable_stats
 version: !ruby/object:Gem::Version 
-  version: 0.0.14
+  version: 0.0.15
 platform: ruby
 authors: 
 - David Richards
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-08-17 00:00:00 -07:00
+date: 2009-08-19 00:00:00 -07:00
 default_executable: jes
 dependencies: []