nuggets 0.9.9 → 1.0.0

data/spec/nuggets/array/mean_spec.rb

@@ -0,0 +1,203 @@
+require 'nuggets/array/mean'
+require 'nuggets/array/standard_deviation'
+
+describe_extended Array, Nuggets::Array::MeanMixin do
+
+  describe 'equalities' do
+
+    %w[arithmetic geometric harmonic].each { |type|
+      method = "#{type}_mean"
+
+      example do
+        val = rand
+        ary = [val] * 10
+
+        ary.send(method).should equal_float(val)
+      end
+
+      example do
+        ary, const = Array.new(10) { rand }, rand
+
+        res1 = ary.map { |i| const * i }.send(method)
+        res2 = ary.send(method)
+
+        res1.should equal_float(const * res2)
+      end
+    }
+
+  end
+
+  describe 'inequalities' do
+
+    before :each do
+      @ary = Array.new(10) { rand }
+    end
+
+    3.times {
+
+      { 0 => 1, 1 => 2, 3 => 23, 0.1 => 0.2, -1 => 0, -3 => -2 }.each { |exponent1, exponent2|
+        example do
+          @ary.generalized_mean(exponent1).should <= @ary.generalized_mean(exponent2)
+        end
+      }
+
+      example do
+        res = %w[quadratic arithmetic geometric harmonic].map { |type| @ary.send("#{type}_mean") }
+        res[0].should >= res[1]; res[1].should >= res[2]; res[2].should >= res[3]
+      end
+
+    }
+
+  end
+
+  describe 'arithmetic mean' do
+
+    example do
+      [].arithmetic_mean.should be_nil
+    end
+
+    example do
+      [1].arithmetic_mean.should == 1.0
+    end
+
+    example do
+      [1, 1, 1].arithmetic_mean.should == 1.0
+    end
+
+    example do
+      [1, 2, 3].arithmetic_mean.should == 2.0
+    end
+
+    example do
+      [3, 2, 1].arithmetic_mean.should == 2.0
+    end
+
+    example do
+      [-3, -2, -1].arithmetic_mean.should == -[3, 2, 1].arithmetic_mean
+    end
+
+    example do
+      [1e16, 1, 3, -1e16].arithmetic_mean.should == 1.0
+    end
+
+    xexample do
+      [1e30, 1, 3, -1e30].arithmetic_mean.should == 1.0
+    end
+
+    context do
+
+      before :each do
+        @ary = [1, -2, 1, 2, 3, -4, 0, 3, 1, 2, 1, 0, 24]
+        @res = @ary.arithmetic_mean
+      end
+
+      example do
+        @res.should equal_float(2.46153846153846)
+      end
+
+      %w[mean average avg].each { |method|
+        example { @ary.send(method).should == @res }
+      }
+
+    end
+
+    describe '(weighted)' do
+
+      example do
+        [1, 2, 3].arithmetic_mean { |i| 0 }.should be_nil
+      end
+
+      example do
+        [1, 2, 3].arithmetic_mean { |i| 1 / i.to_f }.should equal_float(1.63636363636364)
+      end
+
+      example do
+        w = rand
+        [1, 2, 3].arithmetic_mean { |i| w }.should == 2.0
+      end
+
+      example do
+        [1, 2, 3].arithmetic_mean { |i| i ** 2 }.should equal_float(2.57142857142857)
+      end
+
+    end
+
+  end
+
+  # TODO: other methods
+  # TODO: more examples: http://people.revoledu.com/kardi/tutorial/BasicMath/Average/mean.html
+
+  describe 'report mean' do
+
+    before :each do
+      @ary = [[9.4, 34.75], [9.46, 34.68], [9.51, 34.61]]
+
+      @ary1 = @ary.map(&:first)
+      @ary2 = @ary.map(&:last)
+    end
+
+    describe 'w/o stddev' do
+
+      before :each do
+        [@ary, @ary1, @ary2].each { |ary|
+          class << ary; undef_method :std; end if ary.respond_to?(:std)
+        }
+      end
+
+      example do
+        @ary.report_mean.should == ['9.4567', '34.6800']
+      end
+
+      example do
+        @ary.report_mean(:harmonic).should == ['9.4565', '34.6799']
+      end
+
+      example do
+        @ary.unshift(%w[a b]).report_mean(nil, 2).should == ['a  9.46', 'b  34.68']
+      end
+
+      example do
+        @ary1.report_mean.should == ['9.4567']
+      end
+
+      example do
+        @ary2.report_mean.should == ['34.6800']
+      end
+
+      example do
+        [].report_mean.should be_nil
+      end
+
+    end
+
+    describe 'w/ stddev' do
+
+      example do
+        @ary.report_mean.should == ['9.4567 +/- 0.0450', '34.6800 +/- 0.0572']
+      end
+
+      example do
+        @ary.report_mean(:harmonic).should == ['9.4565 +/- 0.0450', '34.6799 +/- 0.0572']
+      end
+
+      example do
+        @ary.unshift(%w[a b]).report_mean(nil, 2).should == ['a  9.46 +/- 0.04', 'b  34.68 +/- 0.06']
+      end
+
+      example do
+        @ary1.report_mean.should == ['9.4567 +/- 0.0450']
+      end
+
+      example do
+        @ary2.report_mean.should == ['34.6800 +/- 0.0572']
+      end
+
+      example do
+        [].report_mean.should be_nil
+      end
+
+    end
+
+  end
+
+end

data/spec/nuggets/array/median_spec.rb

@@ -0,0 +1,77 @@
+require 'nuggets/array/median'
+
+describe_extended Array, Nuggets::Array::MedianMixin do
+
+  example do
+    [].median.should be_nil
+  end
+
+  example do
+    [1].median.should == 1
+  end
+
+  example do
+    [1, 1, 1].median.should == 1
+  end
+
+  example do
+    [1, 2, 3].median.should == 2
+  end
+
+  example do
+    [3, 2, 1].median.should == 2
+  end
+
+  example do
+    [-3, -2, -1].median.should == -[3, 2, 1].median
+  end
+
+  example do
+    [1, -2, 1, 2, 3, -4, 0, 3, 1, 2, 1, 0, 24].median.should == 1
+  end
+
+  example do
+    [1, 2, 3, 4].median.should == 2.5
+  end
+
+  example do
+    %w[one two three].median.should == 'three'
+  end
+
+  example do
+    %w[three two one].median.should == 'three'
+  end
+
+  [true, 1, :left].each { |prefer|
+    example { [1, 2, 3].median(prefer).should == 2 }
+    example { [1, 2, 3, 4].median(prefer).should == 2 }
+
+    example { %w[one two three].median(prefer).should == 'three' }
+    example { %w[one two three four].median(prefer).should == 'one' }
+  }
+
+  [false, 2, :right].each { |prefer|
+    example { [1, 2, 3].median(prefer).should == 2 }
+    example { [1, 2, 3, 4].median(prefer).should == 3 }
+
+    example { %w[one two three].median(prefer).should == 'three' }
+    example { %w[one two three four].median(prefer).should == 'three' }
+  }
+
+  example do
+    [1, 2, 3, 4].median { |a, b| a * b }.should == 6
+  end
+
+  example do
+    %w[one two three four].median { |a, b| b * a.length }.should == 'threethreethree'
+  end
+
+  example do
+    lambda { %w[one two three four].median }.should raise_error(NoMethodError, %r{undefined method `/' .*String})
+  end
+
+  example do
+    lambda { [1, 2, 'three'].median }.should raise_error(ArgumentError, /comparison .* failed/)
+  end
+
+end

data/spec/nuggets/array/mode_spec.rb

@@ -0,0 +1,57 @@
+require 'nuggets/array/mode'
+
+describe_extended Array, Nuggets::Array::ModeMixin do
+
+  example do
+    [].mode.should be_nil
+  end
+
+  example do
+    [1].mode.should == 1
+  end
+
+  example do
+    [1, 1, 1].mode.should == 1
+  end
+
+  example do
+    [1, 2, 2, 3].mode.should == 2
+  end
+
+  example do
+    [3, 2, 2, 1].mode.should == 2
+  end
+
+  example do
+    [-3, -2, -2, -1].mode.should == -[3, 2, 2, 1].mode
+  end
+
+  example do
+    [1, -2, 1, 2, 3, -4, 0, 3, 1, 2, 1, 0, 24].mode.should == 1
+  end
+
+  example do
+    [1, 2, 2, 3, 3, 4].modes.should == [2, 3]
+  end
+
+  example do
+    [2, 3].should include([1, 2, 2, 3, 3, 4].mode)
+  end
+
+  example do
+    %w[one two three three].mode.should == 'three'
+  end
+
+  example do
+    %w[three two one three].mode.should == 'three'
+  end
+
+  example do
+    [1, 2, 2, 3, 4].mode { |x| x * 2 }.should == 4
+  end
+
+  example do
+    [1, 2, 3, 4].modes { |x| x / 2 }.should == [1]
+  end
+
+end

data/spec/nuggets/array/monotone_spec.rb

@@ -0,0 +1,30 @@
+require 'nuggets/array/monotone'
+
+describe Array, 'monotone' do
+
+  example {
+    a = [1, 2, 3, 4]
+
+    a.should be_monotonic
+    a.should be_ascending
+    a.should be_strictly_ascending
+    a.should_not be_descending
+  }
+
+  example {
+    b = [1, 2, 4, 3]
+
+    b.should_not be_monotonic
+    b.should_not be_ascending
+    b.should_not be_descending
+  }
+
+  example {
+    c = [1, 2, 4, 4]
+
+    c.should be_monotonic
+    c.should be_ascending
+    c.should_not be_strictly_ascending
+  }
+
+end

data/spec/nuggets/array/only_spec.rb

@@ -0,0 +1,26 @@
+require 'nuggets/array/only'
+
+describe Array, 'only' do
+
+  example {
+    a = [5]
+
+    a.only.should == 5
+    a.only(true).should == 5
+  }
+
+  example {
+    a = [1, 2, 3]
+
+    lambda { a.only }.should raise_error(IndexError, 'not a single-element array')
+    a.only(true).should == 1
+  }
+
+  example {
+    a = []
+
+    lambda { a.only }.should raise_error(IndexError, 'not a single-element array')
+    a.only(true).should be_nil
+  }
+
+end

data/spec/nuggets/array/regression_spec.rb

@@ -0,0 +1,54 @@
+require 'nuggets/array/regression'
+
+describe_extended Array, Nuggets::Array::RegressionMixin do
+
+  example {
+    [].linear_least_squares.should == []
+  }
+
+  example {
+    [0].linear_least_squares.should == [[1, 0]]
+  }
+
+  example {
+    [1].linear_least_squares.should == [[1, 1]]
+  }
+
+  example {
+    [1, 1].linear_least_squares.should == [[1, 1], [2, 1]]
+  }
+
+  example {
+    [1, 2, 3, 4].linear_least_squares.should == [[1, 1], [2, 2], [3, 3], [4, 4]]
+  }
+
+  example {
+    [1, 2, 4, 1].linear_least_squares.should == [[1, 1.7], [2, 1.9], [3, 2.1], [4, 2.3]]
+  }
+
+  example {
+    r = [2.3, 1.6, 0.9, 0.2]
+
+    [1, 2, 4, -2].linear_least_squares.each_with_index { |(x, y), i|
+      x.should == i + 1
+      y.should equal_float(r[i])
+    }
+  }
+
+  [
+    [],
+    [0],
+    [1, 1],
+    [1, 2, 4, -2],
+    [2.3, 1.6, 0.9, 0.2],
+    [1, 2, -1, 3, -2, 4, -3]
+  ].each { |a|
+    example {
+      a.llsq.zip(a.llsqi.to_a).each { |(x1, y1), (x2, y2)|
+        x1.should == x2
+        y1.should equal_float(y2)
+      }
+    }
+  }
+
+end