nuggets 0.9.9 → 1.0.0

data/spec/nuggets/array/boost_spec.rb

@@ -0,0 +1,50 @@
+require 'nuggets/array/boost'
+
+describe_extended Array, Nuggets::Array::BoostMixin do
+
+  example {
+    lambda { [].boost_factor([]) }.should raise_error(NoMethodError)
+  }
+
+  example {
+    lambda { [1].boost_factor([]) }.should raise_error(NoMethodError)
+  }
+
+  example {
+    lambda { [].boost_factor([1]) }.should raise_error(TypeError)
+  }
+
+  example {
+    a, b = [0].boof([0]), Float::NAN
+
+    a.class.should == b.class
+    a.to_s.should == b.to_s
+  }
+
+  example {
+    1.step(100, 10) { |i| [i].boof([0]).should == -1 }
+  }
+
+  example {
+    1.step(100, 10) { |i| [0].boof([i]).should == Float::INFINITY }
+  }
+
+  example {
+    a, b = [1.5, 1.6, 1.4], [0.7, 0.8, 0.8]
+
+    a.boof(b).should equal_float(-0.48888888888888893)
+    a.boof(b, &:max).should equal_float(-0.5)
+    a.boof(b, &:geomean).should equal_float(-0.4891270450709646)
+    a.boof(b) { |x| x.inject(:-) }.should equal_float(-0.3999999999999999)
+  }
+
+  [
+    [[1], [1], :mean],
+    [[1], [1, 2, 3], :mean],
+    [[1.5, 1.6, 1.4], [0.7, 0.8, 0.8], :mean],
+    [[1.5, 1.6, 1.4], [0.7, 0.8, 0.8], :geomean]
+  ].each { |a, b, m|
+    example { a.boost(a.boof(b, &m)).send(m).should equal_float(b.send(m)) }
+  }
+
+end

data/spec/nuggets/array/combination_spec.rb

@@ -0,0 +1,25 @@
+require 'nuggets/array/combination'
+
+describe Array, 'combination' do
+
+  let(:a) { %w[a b c d] }
+
+  example {
+    r = [%w[a b c], %w[a b d], %w[a c d], %w[b c d]]
+
+    a.comb(3).should == r
+    lambda { |b| a.comb(3, &b) }.should yield_successive_args(*r)
+  }
+
+  example {
+    a.comb(4, 2, 4).should == [%w[a b c d], %w[a b], %w[a c], %w[a d], %w[b c], %w[b d], %w[c d], %w[a b c d]]
+  }
+
+  example {
+    r = [%w[a b c d], %w[a b c], %w[a b d], %w[a c d], %w[b c d], %w[a b], %w[a c], %w[a d], %w[b c], %w[b d], %w[c d], %w[a], %w[b], %w[c], %w[d], []]
+
+    a.comb.should == r
+    lambda { |b| a.comb(&b) }.should yield_successive_args(*r)
+  }
+
+end

data/spec/nuggets/array/correlation_spec.rb

@@ -0,0 +1,81 @@
+require 'nuggets/array/correlation'
+
+describe_extended Array, Nuggets::Array::CorrelationMixin do
+
+  example do
+    [].correlation_coefficient.should == 0.0
+  end
+
+  example do
+    [1].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [1, 1, 1].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [1, 2, 3].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [-1, -2, -3].correlation_coefficient.should == -1.0
+  end
+
+  example do
+    [1, 1.5, 2].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [3, 2, 1].correlation_coefficient.should == -1.0
+  end
+
+  example do
+    [-3, -2, -1].correlation_coefficient.should == -[3, 2, 1].correlation_coefficient
+  end
+
+  example do
+    [1, -2, 1, 2, 3, -4, 0, 3, 1, 2, 1, 0, 24].correlation_coefficient.should equal_float(0.478471417102228)
+  end
+
+  example do
+    [[1, 1]].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [[1, 1], [1, 1], [1, 1]].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [[-1, 1], [-1, 1], [-1, 1]].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [[1, -1], [1, -1], [1, -1]].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [[1, 1], [1, 2], [1, 3]].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [[1, 1], [2, 1], [3, 1]].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [[-1, 1], [-2, 1], [-3, 1]].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [[1, -1], [2, -1], [3, -1]].correlation_coefficient.should == 1.0
+  end
+
+  example do
+    [[3, 1], [2, 2], [1, 3]].correlation_coefficient.should == -1.0
+  end
+
+  example do
+    [[-3, 3], [-2, 2], [-1, 1]].correlation_coefficient.should == -[[3, 3], [2, 2], [1, 1]].correlation_coefficient
+  end
+
+end

data/spec/nuggets/array/flatten_once_spec.rb

@@ -0,0 +1,16 @@
+require 'nuggets/array/flatten_once'
+
+describe Array, 'flatten_once' do
+
+  let(:a) { [1, 2, [3, 4, 5], 6, [7, [8, 9]]] }
+
+  example {
+    a.flatten_once.should == [1, 2, 3, 4, 5, 6, 7, [8, 9]]
+  }
+
+  example {
+    a.flatten_once!
+    a.should == [1, 2, 3, 4, 5, 6, 7, [8, 9]]
+  }
+
+end

data/spec/nuggets/array/flush_spec.rb

@@ -0,0 +1,43 @@
+require 'nuggets/array/flush'
+
+describe_extended Array, Nuggets::Array::FlushMixin do
+
+  example do
+    a = [1, 2, 3, 4]
+    a.flush.should == [1, 2, 3, 4]
+    a.should be_empty
+  end
+
+  example do
+    a = [1, 2, 3, 4]
+    a.flush.push(42).should == [1, 2, 3, 4, 42]
+    a.should be_empty
+  end
+
+  example do
+    a = [1, 2, 3, 4]
+    a.flush { |b| b.should == [1, 2, 3, 4]; 42 }.should == 42
+    a.should be_empty
+  end
+
+  example do
+    a = [1, 2, 3, 4]
+    a.flush { |b| b }.should be_empty
+    a.should be_empty
+  end
+
+  example do
+    a, c = [1, 2, 3, 4], nil
+    a.flush { |b| c = b; c.should == [1, 2, 3, 4] }
+    a.should be_empty
+    c.should be_empty
+  end
+
+  example do
+    a, c = [1, 2, 3, 4], nil
+    a.flush { |b| c = b.dup; c.should == [1, 2, 3, 4] }
+    a.should be_empty
+    c.should == [1, 2, 3, 4]
+  end
+
+end

data/spec/nuggets/array/format_spec.rb

@@ -0,0 +1,52 @@
+require 'nuggets/array/format'
+
+describe Array, 'format' do
+
+  def self.test_format(*f)
+    f.pop.each { |x, y| example { (f.dup % x).should == y } }
+  end
+
+  test_format('"%s"',
+    []       => '',
+    'string' => '"string"',
+    ''       => ''
+  )
+
+  test_format('%s, (%s)', '%s', '(%s)',
+    ['place', 'country'] => 'place, (country)',
+    ['place', ''       ] => 'place',
+    ['',      'country'] => '(country)',
+    ['',      ''       ] => ''
+  )
+
+  test_format('%s: %s (%s)', '%s: %s', '%s (%s)', '%s (%s)',
+    ['author', 'title', 'year'] => 'author: title (year)',
+    ['author', 'title', ''    ] => 'author: title',
+    ['author', '',      'year'] => 'author (year)',
+    ['',       'title', 'year'] => 'title (year)',
+    ['author', ''     , ''    ] => 'author',
+    ['',       'title', ''    ] => 'title',
+    ['',       '',      'year'] => 'year',
+    ['',       '',      ''    ] => ''
+  )
+
+  test_format({ :sep => ':' },
+    ['1', '2', '3', '4'] => '1:2:3:4',
+    ['1', '2', '3', '' ] => '1:2:3',
+    ['1', '2', '',  '4'] => '1:2:4',
+    ['1', '',  '3', '4'] => '1:3:4',
+    ['',  '2', '3', '4'] => '2:3:4',
+    ['1', '2', '',  '' ] => '1:2',
+    ['1', '',  '3', '' ] => '1:3',
+    ['1', '',  '',  '4'] => '1:4',
+    ['',  '2', '3', '' ] => '2:3',
+    ['',  '2', '',  '4'] => '2:4',
+    ['',  '',  '3', '4'] => '3:4',
+    ['1', '',  '',  '' ] => '1',
+    ['',  '2', '',  '' ] => '2',
+    ['',  '',  '3', '' ] => '3',
+    ['',  '',  '',  '4'] => '4',
+    ['',  '',  '',  '' ] => ''
+  )
+
+end

data/spec/nuggets/array/hashify_spec.rb

@@ -0,0 +1,41 @@
+require 'nuggets/array/hashify'
+
+describe_extended Array, Nuggets::Array::HashifyMixin do
+
+  example do
+    hash = [1, 2, 3, 4].hashify
+    hash.should == { 1 => 1, 2 => 2, 3 => 3, 4 => 4 }
+    hash[42].should be_nil
+  end
+
+  example do
+    [1, 2, 3, 4].hashify { |i| [i, i ** 2] }.should == { 1 => 1, 2 => 4, 3 => 9, 4 => 16 }
+  end
+
+  example do
+    [1, 2, 3, 4].hashify { |i| [i ** 2, i] }.should == { 1 => 1, 4 => 2, 9 => 3, 16 => 4 }
+  end
+
+  example do
+    lambda { [1, 2, 3, 4].hashify { |i| i } }.should raise_error(ArgumentError)
+  end
+
+  example do
+    hash = [1, 2, 3, 4].hashify(0)
+    hash.should == { 1 => 1, 2 => 2, 3 => 3, 4 => 4 }
+    hash[42].should == 0
+  end
+
+  example do
+    hash = [1, 2, 3, 4].hashify(lambda { |h, k| h[k] = [] })
+    hash.should == { 1 => 1, 2 => 2, 3 => 3, 4 => 4 }
+    hash[42].should == []
+  end
+
+  example do
+    hash = [1, 2, 3, 4].hashify(4 => -1, 23 => 42)
+    hash.should == { 1 => 1, 2 => 2, 3 => 3, 4 => 4, 23 => 42 }
+    hash[42].should be_nil
+  end
+
+end

data/spec/nuggets/array/histogram_spec.rb

@@ -0,0 +1,87 @@
+require 'nuggets/array/histogram'
+
+describe_extended Array, Nuggets::Array::HistogramMixin do
+
+  example do
+    [].histogram.should == {}
+  end
+
+  example do
+    [1].histogram.should == { 1 => 1 }
+  end
+
+  example do
+    [1, 1, 1].histogram.should == { 1 => 3 }
+  end
+
+  example do
+    [1, 2, 2, 3].histogram.should == { 1 => 1, 2 => 2, 3 => 1 }
+  end
+
+  example do
+    [3, 2, 2, 1].histogram.should == [1, 2, 2, 3].histogram
+  end
+
+  example do
+    [1, -2, 1, 2, 3, -4, 0, 3, 1, 2, 1, 0, 24].histogram.should == {
+      -4 => 1, -2 => 1, 0 => 2, 1 => 4, 2 => 2, 3 => 2, 24 => 1
+    }
+  end
+
+  example do
+    %w[one two three three].histogram.should == { 'one' => 1, 'two' => 1, 'three' => 2 }
+  end
+
+  example do
+    %w[three two one three].histogram.should == %w[one two three three].histogram
+  end
+
+  example do
+    [1, 2, 2, 3, 4].histogram { |x| x * 2 }.should == { 2 => 1, 4 => 2, 6 => 1, 8 => 1 }
+  end
+
+  example do
+    [1, 2, 3, 4].histogram { |x| x / 2 }.should == { 0 => 1, 1 => 2, 2 => 1 }
+  end
+
+  example do
+    [].probability_mass_function.should == {}
+  end
+
+  example do
+    [1].probability_mass_function.should == { 1 => 1 }
+  end
+
+  example do
+    [1, 1, 1].probability_mass_function.should == { 1 => 1 }
+  end
+
+  example do
+    [1, 2, 2, 3].probability_mass_function.should == { 1 => 0.25, 2 => 0.5, 3 => 0.25 }
+  end
+
+  example do
+    [3, 2, 2, 1].probability_mass_function.should == [1, 2, 2, 3].probability_mass_function
+  end
+
+  example do
+    [1, -2, 1, 2, 3, -4, 0, 3, 1, 2, 1, 0, 24].probability_mass_function.values.inject(:+).should equal_float(1.0)
+  end
+
+  example do
+    %w[one two three three].probability_mass_function.should == { 'one' => 0.25, 'two' => 0.25, 'three' => 0.5 }
+  end
+
+  example do
+    %w[three two one three].probability_mass_function.should == %w[one two three three].probability_mass_function
+  end
+
+  example do
+    [1, 2, 2, 3, 4].probability_mass_function { |x| x * 2 }.should == { 2 => 0.2, 4 => 0.4, 6 => 0.2, 8 => 0.2 }
+  end
+
+  example do
+    [1, 2, 3, 4].probability_mass_function { |x| x / 2 }.should == { 0 => 0.25, 1 => 0.5, 2 => 0.25 }
+  end
+
+end

data/spec/nuggets/array/in_order_spec.rb

@@ -0,0 +1,13 @@
+require 'nuggets/array/in_order'
+
+describe Array, 'in_order' do
+
+  let(:a) { [:created_at, :email, :login, :updated_at] }
+
+  example { a.in_order(:login, :email).should == [:login, :email, :created_at, :updated_at] }
+
+  example { a.in_order(:email, :address).should == [:email, :created_at, :login, :updated_at] }
+
+  example { a.in_order!(:login, :email); a.should == [:login, :email, :created_at, :updated_at] }
+
+end

data/spec/nuggets/array/limit_spec.rb

@@ -0,0 +1,62 @@
+require 'nuggets/array/limit'
+require 'nuggets/numeric/limit'
+
+describe_extended Array, Nuggets::Array::LimitMixin do
+
+  describe '#limit' do
+
+    example do
+      [].limit(0, 1).should == []
+    end
+
+    example do
+      [1].limit(0, 1).should == [1]
+    end
+
+    example do
+      [1, 1, 1].limit(0, 1).should == [1]
+    end
+
+    example do
+      [1, 2, 3].limit(0, 1).should == [1]
+    end
+
+    example do
+      [1, 2, 3].limit(0, 2).should == [1, 2]
+    end
+
+    example do
+      [-3, -2, -1].limit(0, 1).should == [0]
+    end
+
+    example do
+      [1, -2, 1, 2, 3, -4, 0, 3, 1, 2, 1, 0, 24].limit(-2, 2).should == [1, -2, 2, 0]
+    end
+
+  end
+
+  describe '#cap' do
+
+    example do
+      [1, 1, 1].cap(1).should == [1, 1, 1]
+    end
+
+    example do
+      [1, 2, 3].cap(1).should == [1, 1, 1]
+    end
+
+    example do
+      [1, 2, 3].cap(2).should == [1, 2, 2]
+    end
+
+    example do
+      [-3, -2, -1].cap(0).should == [-3, -2, -1]
+    end
+
+    example do
+      [1, -2, 1, 2, 3, -4, 0, 3, 1, 2, 1, 0, 24].cap(-2..2).should == [1, -2, 1, 2, 2, -2, 0, 2, 1, 2, 1, 0, 2]
+    end
+
+  end
+
+end