hamster 0.4.3 → 1.0.0

data/spec/lib/hamster/vector/permutation_spec.rb

@@ -0,0 +1,92 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#permutation" do
+    let(:vector) { Hamster.vector(1,2,3,4) }
+
+    context "without a block or arguments" do
+      it "returns an Enumerator of all permutations" do
+        vector.permutation.class.should be(Enumerator)
+        vector.permutation.to_a.should eql(vector.to_a.permutation.to_a)
+      end
+    end
+
+    context "without a block, but with integral argument" do
+      it "returns an Enumerator of all permutations of given length" do
+        vector.permutation(2).class.should be(Enumerator)
+        vector.permutation(2).to_a.should eql(vector.to_a.permutation(2).to_a)
+        vector.permutation(3).class.should be(Enumerator)
+        vector.permutation(3).to_a.should eql(vector.to_a.permutation(3).to_a)
+      end
+    end
+
+    context "with a block" do
+      it "returns self" do
+        vector.permutation {}.should be(vector)
+      end
+
+      context "and no argument" do
+        it "yields all permutations" do
+          yielded = []
+          vector.permutation { |obj| yielded << obj }
+          yielded.sort.should eql([[1,2,3,4], [1,2,4,3], [1,3,2,4], [1,3,4,2],
+            [1,4,2,3], [1,4,3,2], [2,1,3,4], [2,1,4,3], [2,3,1,4], [2,3,4,1],
+            [2,4,1,3], [2,4,3,1], [3,1,2,4], [3,1,4,2], [3,2,1,4], [3,2,4,1],
+            [3,4,1,2], [3,4,2,1], [4,1,2,3], [4,1,3,2], [4,2,1,3], [4,2,3,1],
+            [4,3,1,2], [4,3,2,1]])
+        end
+      end
+
+      context "and an integral argument" do
+        it "yields all permutations of the given length" do
+          yielded = []
+          vector.permutation(2) { |obj| yielded << obj }
+          yielded.sort.should eql([[1,2], [1,3], [1,4], [2,1], [2,3], [2,4], [3,1],
+            [3,2], [3,4], [4,1], [4,2], [4,3]])
+        end
+      end
+    end
+
+    context "on an empty vector" do
+      it "yields the empty permutation" do
+        yielded = []
+        Hamster.vector.permutation { |obj| yielded << obj }
+        yielded.should eql([[]])
+      end
+    end
+
+    context "with an argument of zero" do
+      it "yields the empty permutation" do
+        yielded = []
+        vector.permutation(0) { |obj| yielded << obj }
+        yielded.should eql([[]])
+      end
+    end
+
+    context "with a length greater than the size of the vector" do
+      it "yields no permutations" do
+        vector.permutation(5) { |obj| fail }
+      end
+    end
+
+    it "handles duplicate elements correctly" do
+      Hamster.vector(1,2,3,1).permutation(2).sort.should eql([[1,1], [1,1], [1,2], [1,2],
+        [1,3], [1,3], [2,1],[2,1],[2,3], [3,1],[3,1],[3,2]])
+    end
+
+    it "leaves the original unmodified" do
+      vector.permutation(2) {}
+      vector.should eql(Hamster.vector(1,2,3,4))
+    end
+
+    it "behaves like Array#permutation" do
+      10.times do
+        array  = rand(8).times.map { rand(10000) }
+        vector = V.new(array)
+        perm_size = array.size == 0 ? 0 : rand(array.size)
+        array.permutation(perm_size).to_a.should == vector.permutation(perm_size).to_a
+      end
+    end
+  end
+end

data/spec/lib/hamster/vector/pop_spec.rb

@@ -0,0 +1,27 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#pop" do
+    [
+      [[], []],
+      [["A"], []],
+      [%w[A B C], %w[A B]],
+      [1..32, 1..31],
+      [1..33, 1..32]
+    ].each do |values, expected|
+      context "on #{values.inspect}" do
+        let(:vector) { Hamster.vector(*values) }
+
+        it "preserves the original" do
+          vector.pop
+          vector.should eql(Hamster.vector(*values))
+        end
+
+        it "returns #{expected.inspect}" do
+          vector.pop.should eql(Hamster.vector(*expected))
+        end
+      end
+    end
+  end
+end

data/spec/lib/hamster/vector/product_spec.rb

@@ -0,0 +1,71 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#product" do
+    context "when passed no arguments" do
+      it "multiplies all items in vector" do
+        [
+          [[], 1],
+          [[2], 2],
+          [[1, 3, 5, 7, 11], 1155],
+        ].each do |values, expected|
+          Hamster.vector(*values).product.should == expected
+        end
+      end
+    end
+
+    context "when passed one or more vectors" do
+      let(:vector) { Hamster.vector(1,2,3) }
+
+      context "when passed a block" do
+        it "yields an array for each combination of items from the vectors" do
+          yielded = []
+          vector.product(vector) { |obj| yielded << obj }
+          yielded.should eql([[1,1], [1,2], [1,3], [2,1], [2,2], [2,3], [3,1], [3,2], [3,3]])
+
+          yielded = []
+          vector.product(Hamster.vector(3,4,5), Hamster.vector(6,8)) { |obj| yielded << obj }
+          yielded.should eql(
+            [[1, 3, 6], [1, 3, 8], [1, 4, 6], [1, 4, 8], [1, 5, 6], [1, 5, 8],
+             [2, 3, 6], [2, 3, 8], [2, 4, 6], [2, 4, 8], [2, 5, 6], [2, 5, 8],
+             [3, 3, 6], [3, 3, 8], [3, 4, 6], [3, 4, 8], [3, 5, 6], [3, 5, 8]])
+        end
+
+        it "returns self" do
+          vector.product(Hamster.vector) {}.should be(vector)
+          vector.product(Hamster.vector(1,2), Hamster.vector(3)) {}.should be(vector)
+          Hamster.vector.product(vector) {}.should be(Hamster.vector)
+        end
+      end
+
+      context "when not passed a block" do
+        it "returns the cartesian product in an array" do
+          Hamster.vector(1,2).product(Hamster.vector(3,4,5), Hamster.vector(6,8)).should eql(
+            [[1, 3, 6], [1, 3, 8], [1, 4, 6], [1, 4, 8], [1, 5, 6], [1, 5, 8],
+            [2, 3, 6], [2, 3, 8], [2, 4, 6], [2, 4, 8], [2, 5, 6], [2, 5, 8]])
+        end
+      end
+
+      context "when one of the arguments is empty" do
+        it "returns an empty array" do
+          vector.product(Hamster.vector, Hamster.vector(4,5,6)).should eql([])
+        end
+      end
+
+      context "when the receiver is empty" do
+        it "returns an empty array" do
+          Hamster.vector.product(vector, Hamster.vector(4,5,6)).should eql([])
+        end
+      end
+    end
+
+    context "when passed one or more Arrays" do
+      it "also calculates the cartesian product correctly" do
+        Hamster.vector(1,2).product([3,4,5], [6,8]).should eql(
+          [[1, 3, 6], [1, 3, 8], [1, 4, 6], [1, 4, 8], [1, 5, 6], [1, 5, 8],
+          [2, 3, 6], [2, 3, 8], [2, 4, 6], [2, 4, 8], [2, 5, 6], [2, 5, 8]])
+      end
+    end
+  end
+end

data/spec/{hamster → lib/hamster}/vector/reduce_spec.rb

@@ -1,47 +1,25 @@
-require 'spec_helper'
-
-require 'hamster/vector'
+require "spec_helper"
+require "hamster/vector"
 
 describe Hamster::Vector do
-
-  [:reduce, :inject, :fold, :foldr].each do |method|
-
+  [:reduce, :inject].each do |method|
     describe "##{method}" do
-
       [
         [[], 10, 10],
         [[1], 10, 9],
         [[1, 2, 3], 10, 4],
       ].each do |values, initial, expected|
-
         describe "on #{values.inspect}" do
-
-          before do
-            @vector = Hamster.vector(*values)
-          end
+          let(:vector) { Hamster.vector(*values) }
 
           describe "with an initial value of #{initial}" do
-
             describe "and a block" do
-
               it "returns #{expected.inspect}" do
-                @vector.send(method, initial) { |memo, item| memo - item }.should == expected
+                vector.send(method, initial) { |memo, item| memo - item }.should == expected
               end
-
             end
-
-            describe "and no block" do
-
-              it "returns the memo" do
-                @vector.send(method, initial).should == initial
-              end
-
-            end
-
           end
-
         end
-
       end
 
       [
@@ -49,39 +27,30 @@ describe Hamster::Vector do
         [[1], 1],
         [[1, 2, 3], -4],
       ].each do |values, expected|
-
         describe "on #{values.inspect}" do
-
-          before do
-            @vector = Hamster.vector(*values)
-          end
+          let(:vector) { Hamster.vector(*values) }
 
           describe "with no initial value" do
-
             describe "and a block" do
-
               it "returns #{expected.inspect}" do
-                @vector.send(method) { |memo, item| memo - item }.should == expected
-              end
-
-            end
-
-            describe "and no block" do
-
-              it "returns nil" do
-                @vector.send(method).should be_nil
+                vector.send(method) { |memo, item| memo - item }.should == expected
               end
-
             end
-
           end
-
         end
+      end
 
+      describe "with no block and a symbol argument" do
+        it "uses the symbol as the name of a method to reduce with" do
+          Hamster.vector(1, 2, 3).send(method, :+).should == 6
+        end
       end
 
+      describe "with no block and a string argument" do
+        it "uses the string as the name of a method to reduce with" do
+          Hamster.vector(1, 2, 3).send(method, '+').should == 6
+        end
+      end
     end
-
   end
-
-end
+end

data/spec/lib/hamster/vector/reject_spec.rb

@@ -0,0 +1,44 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  [:reject, :delete_if].each do |method|
+    describe "##{method}" do
+      [
+        [[], []],
+        [["A"], ["A"]],
+        [%w[A B C], %w[A B C]],
+        [%w[A b C], %w[A C]],
+        [%w[a b c], []],
+      ].each do |values, expected|
+        describe "on #{values.inspect}" do
+          let(:vector) { Hamster.vector(*values) }
+
+          context "with a block" do
+            it "returns #{expected.inspect}" do
+              vector.send(method) { |item| item == item.downcase }.should eql(Hamster.vector(*expected))
+            end
+          end
+
+          context "without a block" do
+            it "returns an Enumerator" do
+              vector.send(method).class.should be(Enumerator)
+              vector.send(method).each { |item| item == item.downcase }.should eql(Hamster.vector(*expected))
+            end
+          end
+        end
+      end
+
+      it "works with a variety of inputs" do
+        [1, 2, 10, 31, 32, 33, 1023, 1024, 1025].each do |size|
+          [0, 5, 32, 50, 500, 800, 1024].each do |threshold|
+            vector = V.new(1..size)
+            result = vector.send(method) { |item| item > threshold }
+            result.size.should == [size, threshold].min
+            result.should eql(V.new(1..[size, threshold].min))
+          end
+        end
+      end
+    end
+  end
+end

data/spec/lib/hamster/vector/repeated_combination_spec.rb

@@ -0,0 +1,78 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#repeated_combination" do
+    let(:vector) { Hamster.vector(1,2,3,4) }
+
+    context "with no block" do
+      it "returns an Enumerator" do
+        vector.repeated_combination(2).class.should be(Enumerator)
+      end
+    end
+
+    context "with a block" do
+      it "returns self" do
+        vector.repeated_combination(2) {}.should be(vector)
+      end
+    end
+
+    context "with a negative argument" do
+      it "yields nothing and returns self" do
+        result = []
+        vector.repeated_combination(-1) { |obj| result << obj }.should be(vector)
+        result.should eql([])
+      end
+    end
+
+    context "with a zero argument" do
+      it "yields an empty array" do
+        result = []
+        vector.repeated_combination(0) { |obj| result << obj }
+        result.should eql([[]])
+      end
+    end
+
+    context "with a argument of 1" do
+      it "yields each item in the vector, as single-item vectors" do
+        result = []
+        vector.repeated_combination(1) { |obj| result << obj }
+        result.should eql([[1],[2],[3],[4]])
+      end
+    end
+
+    context "on an empty vector, with an argument greater than zero" do
+      it "yields nothing" do
+        result = []
+        Hamster.vector.repeated_combination(1) { |obj| result << obj }
+        result.should eql([])
+      end
+    end
+
+    context "with a positive argument, greater than 1" do
+      it "yields all combinations of the given size (where a single element can appear more than once in a row)" do
+        vector.repeated_combination(2).to_a.should == [[1,1], [1,2], [1,3], [1,4], [2,2], [2,3], [2,4], [3,3], [3,4], [4,4]]
+        vector.repeated_combination(3).to_a.should == [[1,1,1], [1,1,2], [1,1,3], [1,1,4],
+          [1,2,2], [1,2,3], [1,2,4], [1,3,3], [1,3,4], [1,4,4], [2,2,2], [2,2,3],
+          [2,2,4], [2,3,3], [2,3,4], [2,4,4], [3,3,3], [3,3,4], [3,4,4], [4,4,4]]
+        Hamster.vector(1,2,3).repeated_combination(3).to_a.should == [[1,1,1], [1,1,2],
+          [1,1,3], [1,2,2], [1,2,3], [1,3,3], [2,2,2], [2,2,3], [2,3,3], [3,3,3]]
+      end
+    end
+
+    it "leaves the original unmodified" do
+      vector.repeated_combination(2) {}
+      vector.should eql(Hamster.vector(1,2,3,4))
+    end
+
+    it "behaves like Array#repeated_combination" do
+      0.upto(5) do |comb_size|
+        array = 10.times.map { rand(1000) }
+        V.new(array).repeated_combination(comb_size).to_a.should == array.repeated_combination(comb_size).to_a
+      end
+
+      array = 18.times.map { rand(1000) }
+      V.new(array).repeated_combination(2).to_a.should == array.repeated_combination(2).to_a
+    end
+  end
+end

data/spec/lib/hamster/vector/repeated_permutation_spec.rb

@@ -0,0 +1,94 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#repeated_permutation" do
+    let(:vector) { Hamster.vector(1,2,3,4) }
+
+    context "without a block" do
+      context "and without argument" do
+        it "returns an Enumerator of all repeated permutations" do
+          vector.repeated_permutation.class.should be(Enumerator)
+          vector.repeated_permutation.to_a.sort.should eql(vector.to_a.repeated_permutation(4).to_a.sort)
+        end
+      end
+
+      context "with an integral argument" do
+        it "returns an Enumerator of all repeated permutations of the given length" do
+          vector.repeated_permutation(2).class.should be(Enumerator)
+          vector.repeated_permutation(2).to_a.sort.should eql(vector.to_a.repeated_permutation(2).to_a.sort)
+          vector.repeated_permutation(3).class.should be(Enumerator)
+          vector.repeated_permutation(3).to_a.sort.should eql(vector.to_a.repeated_permutation(3).to_a.sort)
+        end
+      end
+    end
+
+    context "with a block" do
+      it "returns self" do
+        vector.repeated_permutation {}.should be(vector)
+      end
+
+      context "on an empty vector" do
+        it "yields the empty permutation" do
+          yielded = []
+          Hamster.vector.repeated_permutation { |obj| yielded << obj }
+          yielded.should eql([[]])
+        end
+      end
+
+      context "with an argument of zero" do
+        it "yields the empty permutation" do
+          yielded = []
+          vector.repeated_permutation(0) { |obj| yielded << obj }
+          yielded.should eql([[]])
+        end
+      end
+
+      context "with no argument" do
+        it "yields all repeated permutations" do
+          yielded = []
+          Hamster.vector(1,2,3).repeated_permutation { |obj| yielded << obj }
+          yielded.sort.should eql([[1,1,1], [1,1,2], [1,1,3], [1,2,1], [1,2,2],
+            [1,2,3], [1,3,1], [1,3,2], [1,3,3], [2,1,1], [2,1,2], [2,1,3], [2,2,1],
+            [2,2,2], [2,2,3], [2,3,1], [2,3,2], [2,3,3], [3,1,1], [3,1,2], [3,1,3],
+            [3,2,1], [3,2,2], [3,2,3], [3,3,1], [3,3,2], [3,3,3]])
+        end
+      end
+
+      context "with a positive integral argument" do
+        it "yields all repeated permutations of the given length" do
+          yielded = []
+          vector.repeated_permutation(2) { |obj| yielded << obj }
+          yielded.sort.should eql([[1,1], [1,2], [1,3], [1,4], [2,1], [2,2], [2,3], [2,4],
+            [3,1], [3,2], [3,3], [3,4], [4,1], [4,2], [4,3], [4,4]])
+        end
+      end
+    end
+
+    it "handles duplicate elements correctly" do
+    Hamster.vector(10,11,10).repeated_permutation(2).sort.should eql([[10, 10], [10, 10],
+      [10, 10], [10, 10], [10, 11], [10, 11], [11, 10], [11, 10], [11, 11]])
+    end
+
+    it "allows permutations larger than the number of elements" do
+      Hamster.vector(1,2).repeated_permutation(3).sort.should eql(
+        [[1, 1, 1], [1, 1, 2], [1, 2, 1],
+         [1, 2, 2], [2, 1, 1], [2, 1, 2],
+         [2, 2, 1], [2, 2, 2]])
+    end
+
+    it "leaves the original unmodified" do
+      vector.repeated_permutation(2) {}
+      vector.should eql(Hamster.vector(1,2,3,4))
+    end
+
+    it "behaves like Array#repeated_permutation" do
+      10.times do
+        array  = rand(8).times.map { rand(10000) }
+        vector = V.new(array)
+        perm_size = array.size == 0 ? 0 : rand(array.size)
+        array.repeated_permutation(perm_size).to_a.should == vector.repeated_permutation(perm_size).to_a
+      end
+    end
+  end
+end