nutella 0.3 → 0.4

data/lib/nutella/core_ext/enumerable.rb

@@ -40,7 +40,8 @@ module Enumerable
     self.replace group(size, discard_excess)
   end
 
-  # Add together all numeric elements in the collection.
+  # Add together all numeric elements in the collection. When using a block,
+  # adds together all numeric elements that pass the predicate in that block.
   #
   # @example Sums up the elements of an array
   #   [1, 2, 3].sum         # => 6
@@ -50,9 +51,15 @@ module Enumerable
   #   { a: 4, b: 5 }.sum    # => 9
   # @example Ignores the non-numeric element in the array
   #   [2, "string", 9].sum  # => 11
+  # @example Sums up the even numbers from 1 to 10
+  #   (1..10).sum &:even?
+  # @example Sums up the numbers from 1 to 100 that are multiples of 3
+  #   (1..100).sum { |n| n % 3 == 0 }
   #
   # @return [Numeric] the sum of all numeric elements in the collection
-  def sum
-    flatten.select { |e| e.is_a? Numeric }.inject(:+) || 0
+  def sum(&block)
+    # TODO: Two selects in a row, how to DRY this up?
+    flat = respond_to?(:flatten) ? flatten : self
+    flat.select { |e| e.is_a? Numeric }.select(&block).inject(:+) || 0
   end
 end

data/lib/nutella/core_ext/integer.rb

@@ -1,5 +1,36 @@
 class Integer
-  # Check whether the integer is evenly divisible by all of the arguments.
+  # Checks whether the integer goes evenly into all of the arguments
+  #
+  # @example Check if 3 goes into some numbers
+  #   3.goes_into? 10  # => false
+  #   3.goes_into? 6   # => true
+  #
+  # @example Check if 5 goes into every number in some sets of numbers
+  #   5.goes_into? 10, 12  # => false
+  #   5.goes_into? 10, 20  # => true
+  #
+  # @param [*Integer] nums the integer(s) to check against
+  # @return [Boolean] whether or not the integer goes evenly into all of the
+  #                   arguments
+  def goes_into?(*nums)
+    nums.all? { |n| (!zero? && n % self == 0) || n.zero? }
+  end
+
+  # Checks whether the integer goes evenly into any of the arguments
+  #
+  # @example Check if 5 goes into every number in some sets of numbers
+  #   5.goes_into_any? 10, 20  # => true
+  #   5.goes_into_any? 10, 12  # => true
+  #   5.goes_into_any? 12, 21  # => false
+  #
+  # @param [*Integer] nums the integer(s) to check against
+  # @return [Boolean] whether or not the integer goes evenly into any of the
+  #                   arguments
+  def goes_into_any?(*nums)
+    nums.any? { |n| goes_into? n }
+  end
+
+  # Checks whether the integer is evenly divisible by all of the arguments.
   #
   # @example Check if some numbers are evenly divisible by 5
   #   4.multiple_of? 5   # => false
@@ -17,10 +48,10 @@ class Integer
   end
   alias_method :divisible_by?, :multiple_of?
 
-  # Check whether the integer is evenly divisible by any of the arguments.
+  # Checks whether the integer is evenly divisible by any of the arguments.
   #
   # @example {Project Euler #1}[http://projecteuler.net/problem=1] solution using Nutella
-  #   (1...1000).select { |n| n.multiple_of_any?(3, 5) }.sum
+  #   (1...1000).sum { |n| n.multiple_of_any?(3, 5) }
   #
   # @param [*Integer] nums the integer(s) to check against
   # @return [Boolean] whether or not the integer is evenly divisible by any

data/lib/nutella/version.rb

@@ -1,4 +1,4 @@
 module Nutella
   # The current version of Nutella.
-  VERSION = "0.3"
+  VERSION = "0.4"
 end

data/spec/nutella/core_ext/enumerable_spec.rb

@@ -73,6 +73,7 @@ describe Enumerable do
       [].sum.should == 0
       [5].sum.should == 5
       [1, 2, 3].sum.should == 6
+      (1..4).sum.should == 10
     end
 
     it "should flatten the elements before adding" do
@@ -85,5 +86,17 @@ describe Enumerable do
       [1, "str", 4].sum.should == 5
       ["no numbers"].sum.should == 0
     end
+
+    context "with a block" do
+      it "should return the sum of the elements that pass the filter block" do
+        (1..10).sum(&:even?).should == 30
+        (1..10).sum { |n| n % 3 == 0 }.should == 18
+      end
+
+      it "should still filter out all non-numbers" do
+        { a: 1, b: 2, c: 3 }.sum(&:odd?).should == 4
+        [2, "str", 4, 7].sum(&:even?).should == 6
+      end
+    end
   end
 end

data/spec/nutella/core_ext/integer_spec.rb

@@ -7,6 +7,65 @@ describe Integer do
     test_alias Integer, :divisible_by_any?, :multiple_of_any?
   end
 
+  describe "#goes_into?" do
+    it "should return true if the number goes evenly into the argument" do
+      5.goes_into?(10).should be_true
+      3.goes_into?(21).should be_true
+      25.goes_into?(100).should be_true
+    end
+
+    it "should return false if the number does not go evenly in" do
+      3.goes_into?(10).should be_false
+      9.goes_into?(40).should be_false
+      10.goes_into?(5).should be_false
+    end
+
+    context "when passing in zero" do
+      it "should return false if one tries to divide by zero" do
+        0.goes_into?(20).should be_false
+        0.goes_into?(30).should be_false
+      end
+
+      it "should allow zero to go into zero" do
+        0.goes_into?(0).should be_true
+      end
+    end
+
+    context "with multiple arguments" do
+      it "should return true if all arguments succeed" do
+        5.goes_into?(10, 15, 50).should be_true
+        2.goes_into?(2, 4, 10).should be_true
+      end
+
+      it "should return false if only some arguments succeed" do
+        5.goes_into?(10, 12, 15).should be_false
+        8.goes_into?(4, 16).should be_false
+      end
+
+      it "should return false if no arguments succeed" do
+        3.goes_into?(8, 16, 20).should be_false
+        6.goes_into?(5, 10, 15).should be_false
+      end
+    end
+  end
+
+  describe "#goes_into_any?" do
+    it "should return true if all arguments succeed" do
+      5.goes_into_any?(10, 15, 50).should be_true
+      2.goes_into_any?(2, 4, 10).should be_true
+    end
+
+    it "should return true if only some arguments succeed" do
+      5.goes_into_any?(10, 12, 15).should be_true
+      8.goes_into_any?(4, 16).should be_true
+    end
+
+    it "should return false if no arguments succeed" do
+      3.goes_into_any?(8, 16, 20).should be_false
+      6.goes_into_any?(5, 10, 15).should be_false
+    end
+  end
+
   describe "#multiple_of?" do
     it "should return true if the number is evenly divisible" do
       5.multiple_of?(5).should be_true

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: nutella
 version: !ruby/object:Gem::Version
-  version: '0.3'
+  version: '0.4'
   prerelease: 
 platform: ruby
 authors:
@@ -135,7 +135,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 1233150412608386870
+      hash: -2685001604496901448
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24