roulette_wheel_selection 1.0.0

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.rspec

@@ -0,0 +1,2 @@
+--color
+--format progress

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in roulette_wheel_selection.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Yoshimoto
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,36 @@
+# RouletteWheelSelection
+
+Hash#sample method is implemented with roulette wheel selection algorithm. Probability distribution is given by hash values.
+
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'roulette_wheel_selection'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install roulette_wheel_selection
+
+## Usage
+
+Set a hash for probability each keys. Hash#sample method returns random keys by probability distributions.
+
+  > { a: 50, b: 30, c: 20 }.sample
+  => :a   # 50%
+  => :b   # 30%
+  => :c   # 20%
+
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/init.rb

@@ -0,0 +1 @@
+require "roulette_wheel_selection"

data/lib/roulette_wheel_selection.rb

@@ -0,0 +1,3 @@
+require "roulette_wheel_selection/version"
+require "roulette_wheel_selection/array/roulette_wheel_selection"
+require "roulette_wheel_selection/hash/sample"

data/lib/roulette_wheel_selection/array/roulette_wheel_selection.rb

@@ -0,0 +1,21 @@
+class Array
+  def roulette_wheel_selection n=nil
+    return nil if self.empty? && n.nil?
+    return [] if self.empty?
+
+    n1 = n
+    n1 = 1 if n.nil?
+    n1 = n1.to_int
+  rescue Exception => e
+    raise TypeError, "Coercion error: #{n.inspect}.to_int => Integer failed:\n(#{e.message})"
+  else
+    cumulative = self.inject( [0] ) { |s,i| s << s.last + i.to_i }
+    cumulative.shift
+    result = Array.new(n1).map! {
+      r = rand( cumulative.last )
+      cumulative.find_index { |t| r < t }
+    }
+    return result.first if n.nil?
+    result
+  end
+end

data/lib/roulette_wheel_selection/hash/sample.rb

@@ -0,0 +1,10 @@
+class Hash
+  def roulette_wheel_selection n=nil
+    indexes = self.values.roulette_wheel_selection n
+
+    return nil if indexes.nil?
+    return self.keys[ indexes ] unless Array === indexes
+    self.keys.values_at *indexes
+  end
+  alias :sample :roulette_wheel_selection
+end

data/lib/roulette_wheel_selection/version.rb

@@ -0,0 +1,3 @@
+module RouletteWheelSelection
+  VERSION = "1.0.0"
+end

data/roulette_wheel_selection.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/roulette_wheel_selection/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Beyond"]
+  gem.email         = ["beyond@be.to"]
+  gem.description   = %q{ Hash#sample method is implemented with roulette wheel selection algorithm. Probability distribution is given by hash values. }
+  gem.summary       = %q{ Hash#sample method implementation like Array#sample }
+  gem.homepage      = "https://github.com/beyond/roulette_wheel_selection"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "roulette_wheel_selection"
+  gem.require_paths = ["lib"]
+  gem.version       = RouletteWheelSelection::VERSION
+
+  gem.add_development_dependency "rspec"
+end

data/spec/array/roulette_wheel_selection_spec.rb

@@ -0,0 +1,58 @@
+require 'spec_helper'
+
+describe Array do
+  describe "#roulette_wheel_selection" do
+    context "empty" do
+      subject { Array.new }
+
+      it do
+        subject.roulette_wheel_selection.should == nil
+      end
+
+      it do
+        subject.roulette_wheel_selection(1).should == []
+      end
+
+      it do
+        subject.roulette_wheel_selection(5).should == []
+      end
+    end
+
+    context "one value" do
+      subject { [10] }
+
+      it do
+        subject.roulette_wheel_selection.should == 0
+      end
+
+      it do
+        subject.roulette_wheel_selection(1).should == [0]
+      end
+
+      it do
+        subject.roulette_wheel_selection(2).should == [ 0, 0 ]
+      end
+
+      it do
+        o = Object.new
+        def o.to_int; 1; end
+        subject.roulette_wheel_selection(o).should == [0]
+      end
+
+      it do
+        o = Object.new
+        lambda { subject.roulette_wheel_selection(o) }.should raise_error TypeError
+      end
+
+      it do
+        o = Object.new
+        def o.to_int; ''; end
+        lambda { subject.roulette_wheel_selection(o) }.should raise_error TypeError
+      end
+
+      it do
+        lambda { subject.roulette_wheel_selection -7 }.should raise_error ArgumentError
+      end
+    end
+  end
+end

data/spec/hash/sample_spec.rb

@@ -0,0 +1,92 @@
+require 'spec_helper'
+
+describe Hash do
+  describe "#sample" do
+    context "empty hash" do
+      subject { Hash.new }
+
+      it do
+        subject.sample.should == nil
+      end
+
+      it do
+        subject.sample(1).should == []
+      end
+
+      it do
+        subject.sample(5).should == []
+      end
+    end
+
+    context "one key" do
+      subject { { a: 10 } }
+
+      it do
+        subject.sample.should == :a
+      end
+
+      it do
+        subject.sample(1).should == [:a]
+      end
+
+      it do
+        subject.sample(2).should == [ :a, :a ]
+      end
+
+      it do
+        o = Object.new
+        def o.to_int; 1; end
+        subject.sample(o).should == [:a]
+      end
+
+      it do
+        o = Object.new
+        lambda { subject.sample(o) }.should raise_error TypeError
+      end
+
+      it do
+        o = Object.new
+        def o.to_int; ''; end
+        lambda { subject.sample(o) }.should raise_error TypeError
+      end
+
+      it do
+        lambda { subject.sample -7 }.should raise_error ArgumentError
+      end
+    end
+
+    context "Chi-square test" do
+      before do
+        @h = { a: 10, b: 10, c: 25, d: 1, e: 2, f: 2, g: 5, h: 10, i: 20, j: 15 }
+      end
+
+      def calc_chi2 dist, excepted_freq
+        excepted_freq.inject( 0.0 ) { |r,x|
+          k,v = *x
+          r += ((dist[k] - v) ** 2).to_f / v
+          r
+        }
+      end
+
+      it "should be same probability" do
+        chi2_9 = [
+          14.684, 12.242, 10.656, 9.414, 8.343,
+          7.357,  6.393,  5.380, 4.168, 0.000,
+        ]
+
+        f1 = Hash[ @h.map { |k,x| [ k, 1.0 * x / @h.values.inject(&:+) * 1000 ] } ]
+        count = Hash.new(0)
+        2000.times {
+          stats = @h.sample(1000).inject( Hash.new(0) ) { |h,k| h[k]+=1; h }
+          chi2 = calc_chi2 stats, f1
+          i = chi2_9.find_index { |x| chi2 > x }
+          count[i] += 1
+        }
+        f2 = Hash[ (0..9).map { |x| [ x, 200 ] } ]
+
+        chi2 =  calc_chi2 count, f2
+        chi2.should < 21.666    # 1%
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,19 @@
+require File.expand_path("../init", File.dirname(__FILE__))
+
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper"` to ensure that it is only
+# loaded once.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+RSpec.configure do |config|
+  config.treat_symbols_as_metadata_keys_with_true_values = true
+  config.run_all_when_everything_filtered = true
+  config.filter_run :focus
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = 'random'
+end

metadata

@@ -0,0 +1,75 @@
+--- !ruby/object:Gem::Specification
+name: roulette_wheel_selection
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Beyond
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-08-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &9862680 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *9862680
+description: ! ' Hash#sample method is implemented with roulette wheel selection algorithm.
+  Probability distribution is given by hash values. '
+email:
+- beyond@be.to
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- init.rb
+- lib/roulette_wheel_selection.rb
+- lib/roulette_wheel_selection/array/roulette_wheel_selection.rb
+- lib/roulette_wheel_selection/hash/sample.rb
+- lib/roulette_wheel_selection/version.rb
+- roulette_wheel_selection.gemspec
+- spec/array/roulette_wheel_selection_spec.rb
+- spec/hash/sample_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/beyond/roulette_wheel_selection
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Hash#sample method implementation like Array#sample
+test_files:
+- spec/array/roulette_wheel_selection_spec.rb
+- spec/hash/sample_spec.rb
+- spec/spec_helper.rb