hamster 1.0.1.pre.rc2 → 1.0.1.pre.rc3

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

@@ -0,0 +1,57 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  [
+    [:sort, ->(left, right) { left.length <=> right.length }],
+    [:sort_by, ->(item) { item.length }],
+  ].each do |method, comparator|
+    describe "##{method}" do
+      [
+        [[], []],
+        [["A"], ["A"]],
+        [%w[Ichi Ni San], %w[Ni San Ichi]],
+      ].each do |values, expected|
+        describe "on #{values.inspect}" do
+          let(:vector) { Hamster.vector(*values) }
+
+          context "with a block" do
+            it "preserves the original" do
+              vector.send(method, &comparator)
+              vector.should eql(Hamster.vector(*values))
+            end
+
+            it "returns #{expected.inspect}" do
+              vector.send(method, &comparator).should eql(Hamster.vector(*expected))
+            end
+          end
+
+          context "without a block" do
+            it "preserves the original" do
+              vector.send(method)
+              vector.should eql(Hamster.vector(*values))
+            end
+
+            it "returns #{expected.sort.inspect}" do
+              vector.send(method).should eql(Hamster.vector(*expected.sort))
+            end
+          end
+        end
+      end
+
+      [10, 31, 32, 33, 1023, 1024, 1025].each do |size|
+        context "on a #{size}-item vector" do
+          it "behaves like Array#{method}" do
+            array = size.times.map { rand(10000) }
+            vector = V.new(array)
+            if method == :sort
+              vector.sort.should == array.sort
+            else
+              vector.sort_by { |x| -x }.should == array.sort_by { |x| -x }
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/spec/{hamster/set → lib/hamster/vector}/sum_spec.rb

@@ -1,22 +1,16 @@
 require "spec_helper"
-require "hamster/set"
+require "hamster/vector"
 
-describe Hamster::Set do
+describe Hamster::Vector do
   describe "#sum" do
     [
       [[], 0],
       [[2], 2],
       [[1, 3, 5, 7, 11], 27],
     ].each do |values, expected|
-
       describe "on #{values.inspect}" do
-        before do
-          original = Hamster.set(*values)
-          @result = original.sum
-        end
-
         it "returns #{expected.inspect}" do
-          @result.should == expected
+          Hamster.vector(*values).sum.should == expected
         end
       end
     end

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

@@ -0,0 +1,29 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#take" do
+    [
+      [[], 10, []],
+      [["A"], 10, ["A"]],
+      [%w[A B C], 0, []],
+      [%w[A B C], 2, %w[A B]],
+      [(1..32), 1, [1]],
+      [(1..33), 32, (1..32)],
+      [(1..100), 40, (1..40)]
+    ].each do |values, number, expected|
+      describe "#{number} from #{values.inspect}" do
+        let(:vector) { Hamster.vector(*values) }
+
+        it "preserves the original" do
+          vector.take(number)
+          vector.should eql(Hamster.vector(*values))
+        end
+
+        it "returns #{expected.inspect}" do
+          vector.take(number).should eql(Hamster.vector(*expected))
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,35 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#take_while" do
+    [
+      [[], []],
+      [["A"], ["A"]],
+      [%w[A B C], %w[A B]]
+    ].each do |values, expected|
+      describe "on #{values.inspect}" do
+        let(:vector) { Hamster.vector(*values) }
+        let(:result) { vector.take_while { |item| item < "C" }}
+
+        describe "with a block" do
+          it "returns #{expected.inspect}" do
+            result.should eql(Hamster.vector(*expected))
+          end
+
+          it "preserves the original" do
+            result
+            vector.should eql(Hamster.vector(*values))
+          end
+        end
+
+        describe "without a block" do
+          it "returns an Enumerator" do
+            vector.take_while.class.should be(Enumerator)
+            vector.take_while.each { |item| item < "C" }.should eql(Hamster.vector(*expected))
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -11,33 +11,32 @@ describe Hamster::Vector do
       it "returns the values" do
         expect(to_a).to eq(values)
       end
-
-      it "doesn't end with 'The End'" do
-        expect(to_a).to_not include("The End")
-      end
-
-      it "returns a mutable array" do
-        to_a << "The End"
-        expect(to_a).to include("The End")
-      end
     end
 
-    context "with an empty array" do
+    context "with an empty vector" do
       let(:values) { [] }
 
       include_examples "checking to_a values"
     end
 
-    context "with an single item array" do
+    context "with an single item vector" do
       let(:values) { %w[A] }
 
       include_examples "checking to_a values"
     end
 
-    context "with an multi-item array" do
+    context "with an multi-item vector" do
       let(:values) { %w[A B] }
 
       include_examples "checking to_a values"
     end
+
+    [10, 31, 32, 33, 1000, 1023, 1024, 1025].each do |size|
+      context "with a #{size}-item vector" do
+        let(:values) { (1..size).to_a }
+
+        include_examples "checking to_a values"
+      end
+    end
   end
 end

data/spec/{hamster/set → lib/hamster/vector}/to_list_spec.rb

@@ -1,34 +1,29 @@
 require "spec_helper"
-require "hamster/set"
+require "hamster/vector"
 require "hamster/list"
 
-describe Hamster::Set do
+describe Hamster::Vector do
   describe "#to_list" do
     [
       [],
       ["A"],
       %w[A B C],
     ].each do |values|
-
       describe "on #{values.inspect}" do
-        before do
-          set = Hamster.set(*values)
-          @list = set.to_list
-        end
+        let(:vector) { Hamster::Vector.new(values) }
+        let(:list) { vector.to_list }
 
         it "returns a list" do
-          @list.is_a?(Hamster::List).should == true
+          list.is_a?(Hamster::List).should == true
         end
 
         describe "the returned list" do
           it "has the correct length" do
-            @list.size.should == values.size
+            list.size.should == values.size
           end
 
           it "contains all values" do
-            values.each do |value|
-              @list.should include(value)
-            end
+            list.to_a.should == values
           end
         end
       end

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

@@ -0,0 +1,23 @@
+require "spec_helper"
+require "hamster/vector"
+require "hamster/set"
+
+describe Hamster::Vector do
+  describe "#to_set" do
+    [
+      [],
+      ["A"],
+      %w[A B C],
+      (1..10),
+      (1..32),
+      (1..33),
+      (1..1000)
+    ].each do |values|
+      describe "on #{values.inspect}" do
+        it "returns a set with the same values" do
+          Hamster.vector(*values).to_set.should eql(Hamster.set(*values))
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,49 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#transpose" do
+    it "takes a vector of vectors and transposes rows and columns" do
+      V[V[1, 'a'], V[2, 'b'], V[3, 'c']].transpose.should eql(V[V[1, 2, 3], V["a", "b", "c"]])
+      V[V[1, 2, 3], V["a", "b", "c"]].transpose.should eql(V[V[1, 'a'], V[2, 'b'], V[3, 'c']])
+      V[].transpose.should eql(V[])
+      V[V[]].transpose.should eql(V[])
+      V[V[], V[]].transpose.should eql(V[])
+      V[V[0]].transpose.should eql(V[V[0]])
+      V[V[0], V[1]].transpose.should eql(V[V[0, 1]])
+    end
+
+    it "raises an IndexError if the vectors are not of the same length" do
+      -> { V[V[1,2], V[:a]].transpose }.should raise_error(IndexError)
+    end
+
+    it "also works on Vectors of Arrays" do
+      V[[1,2,3], [4,5,6]].transpose.should eql(V[V[1,4], V[2,5], V[3,6]])
+    end
+
+    [10, 31, 32, 33, 1000, 1023, 1024, 1025, 2000].each do |size|
+      context "on #{size}-item vectors" do
+        it "behaves like Array#transpose" do
+          array = rand(10).times.map { size.times.map { rand(10000) }}
+          vector = V.new(array)
+          result = vector.transpose
+          # Array#== uses Object#== to compare corresponding elements,
+          #   so although Vector#== does type coercion, it does not consider
+          #   nested Arrays and corresponding nested Vectors to be equal
+          # That is why the following ".map { |a| V.new(a) }" is needed
+          result.should == array.transpose.map { |a| V.new(a) }
+          result.each { |v| v.class.should be(Hamster::Vector) }
+        end
+      end
+    end
+
+    context "on a subclass of Vector" do
+      it "returns instances of the subclass" do
+        subclass = Class.new(V)
+        instance = subclass.new([[1,2,3], [4,5,6]])
+        instance.transpose.class.should be(subclass)
+        instance.transpose.each { |v| v.class.should be(subclass) }
+      end
+    end
+  end
+end

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

@@ -0,0 +1,56 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#uniq" do
+    let(:vector) { Hamster.vector('a', 'b', 'a', 'a', 'c', 'b') }
+
+    it "returns a vector with no duplicates" do
+      vector.uniq.should eql(Hamster.vector('a', 'b', 'c'))
+    end
+
+    it "leaves the original unmodified" do
+      vector.uniq
+      vector.should eql(Hamster.vector('a', 'b', 'a', 'a', 'c', 'b'))
+    end
+
+    it "uses #eql? semantics" do
+      Hamster.vector(1.0, 1).uniq.should eql(Hamster.vector(1.0, 1))
+    end
+
+    it "also uses #hash when determining which values are duplicates" do
+      x = double(1)
+      x.should_receive(:hash).at_least(1).times.and_return(1)
+      y = double(2)
+      y.should_receive(:hash).at_least(1).times.and_return(2)
+      Hamster.vector(x, y).uniq
+    end
+
+    it "keeps the first of each group of duplicate values" do
+      x, y, z = 'a', 'a', 'a'
+      result = Hamster.vector(x, y, z).uniq
+      result.size.should == 1
+      result[0].should be(x)
+    end
+
+    [10, 31, 32, 33, 1000, 1023, 1024, 1025, 2000].each do |size|
+      context "on a #{size}-item vector" do
+        it "behaves like Array#uniq" do
+          array = size.times.map { rand(size*2) }
+          vector = V.new(array)
+          result = vector.uniq
+          result.should == array.uniq
+          result.class.should be(Hamster::Vector)
+        end
+      end
+    end
+
+    context "from a subclass" do
+      it "returns an instance of the subclass" do
+        subclass = Class.new(Hamster::Vector)
+        instance = subclass.new([1,2,3])
+        instance.uniq.class.should be(subclass)
+      end
+    end
+  end
+end

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

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

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

@@ -0,0 +1,34 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#values_at" do
+    let(:vector) { Hamster.vector('a', 'b', 'c') }
+
+    it "accepts any number of indices, and returns a vector of items at those indices" do
+      vector.values_at(0).should eql(Hamster.vector('a'))
+      vector.values_at(1,2).should eql(Hamster.vector('b', 'c'))
+    end
+
+    context "when passed invalid indices" do
+      it "fills in with nils" do
+        vector.values_at(1,2,3).should eql(Hamster.vector('b', 'c', nil))
+        vector.values_at(-10,10).should eql(Hamster.vector(nil, nil))
+      end
+    end
+
+    context "when passed no arguments" do
+      it "returns an empty vector" do
+        vector.values_at.should eql(Hamster.vector)
+      end
+    end
+
+    context "from a subclass" do
+      it "returns an instance of the subclass" do
+        subclass = Class.new(Hamster::Vector)
+        instance = subclass.new([1,2,3])
+        instance.values_at(1,2).class.should be(subclass)
+      end
+    end
+  end
+end

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

@@ -0,0 +1,58 @@
+require "spec_helper"
+require "hamster/vector"
+
+describe Hamster::Vector do
+  describe "#zip" do
+    let(:vector) { Hamster.vector(1,2,3,4) }
+
+    context "with a block" do
+      it "yields arrays of one corresponding element from each input sequence" do
+        result = []
+        vector.zip(['a', 'b', 'c', 'd']) { |obj| result << obj }
+        result.should eql([[1,'a'], [2,'b'], [3,'c'], [4,'d']])
+      end
+
+      it "fills in the missing values with nils" do
+        result = []
+        vector.zip(['a', 'b']) { |obj| result << obj }
+        result.should eql([[1,'a'], [2,'b'], [3,nil], [4,nil]])
+      end
+
+      it "returns nil" do
+        vector.zip([2,3,4]) {}.should be_nil
+      end
+
+      it "can handle multiple inputs, of different classes" do
+        result = []
+        vector.zip(V[2,3,4,5], [5,6,7,8]) { |obj| result << obj }
+        result.should eql([[1,2,5], [2,3,6], [3,4,7], [4,5,8]])
+      end
+    end
+
+    context "without a block" do
+      it "returns a vector of arrays (one corresponding element from each input sequence)" do
+        vector.zip([2,3,4,5]).should eql(Hamster.vector([1,2], [2,3], [3,4], [4,5]))
+      end
+    end
+
+    [10, 31, 32, 33, 1000, 1023, 1024, 1025].each do |size|
+      context "on #{size}-item vectors" do
+        it "behaves like Array#zip" do
+          array   = (rand(9)+1).times.map { size.times.map { rand(10000) }}
+          vectors = array.map { |a| V.new(a) }
+          result  = vectors.first.zip(*vectors.drop(1))
+          result.class.should be(Hamster::Vector)
+          result.should == array[0].zip(*array.drop(1))
+        end
+      end
+    end
+
+    context "from a subclass" do
+      it "returns an instance of the subclass" do
+        subclass = Class.new(Hamster::Vector)
+        instance = subclass.new([1,2,3])
+        instance.zip([4,5,6]).class.should be(subclass)
+      end
+    end
+  end
+end