weighted_distribution 1.0.0 → 1.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: c9bb5df8a285b645b8a36b31668dadf1166b55f3
+  data.tar.gz: 8298252b04f543bd69882a143b383c3b02bc9609
+SHA512:
+  metadata.gz: 920a1dcf8422e436c035b6ac1dbd638af3d241112afdf1c68c201f4532a373ae5a26e4b8392552c33efd175decca2292816464c11cc03487e93507f45a5b822b
+  data.tar.gz: c3b551c34ff9ca4bea44abc29f4072fcd0db62390e66e65dd55dbe9b05a1399b93670617ebb893dae765e2fc4db85d533d23cec62143e195cc7690b1d2a29bf7

data/lib/weighted_distribution.rb

@@ -2,8 +2,8 @@
 # Constructor expects a hash of object => weight pairs where the
 # weight represents the probability of the corresponding
 # object being selected. follows specification of ryanlecompte's
-# WeightedDistributionizer[link:https://github.com/ryanlecompte/weighted_randomizer]
-# reimplements the sample function to run in O(log(n)) rather than O(n).
+# WeightedRandomizer[link:https://github.com/ryanlecompte/weighted_randomizer]
+# re-implements the sample function to run in O(log(n)) rather than O(n).
 #
 # @example Usage
 #  suit_dist = WeightedDistribution.new({heart: 12, spade: 11, diamond: 10, club:13})
@@ -12,17 +12,18 @@
 # @note Mostly adapted from recipe 5.11 from the Ruby Cookbook.
 class WeightedDistribution
 
-  # Creates a new instance of the WeightedDistributionizer Class
+  # Creates a new instance of the WeightedDistribution Class
   #
   # @param [Hash] object_weights objects with their corresponding
   # weights(key object, value weight)
-  # @return [WeightedDistributionizer]
-  def initialize(object_weights)
+  # @return [WeightedDistribution]
+  def initialize(object_weights, randomizer = Kernel)
     @len = object_weights.length
     @keys = object_weights.keys
     @orig_values = @keys.map{|x| object_weights[x]}
     @values = normalize(@orig_values)
     @csum = cumulative_sum(@values)
+    @randomizer = randomizer
   end
 
   # Samples from the WeightedDistribution. Each object has a
@@ -45,7 +46,7 @@ class WeightedDistribution
   #
   # @return [Object] sampled object from distribution
   def _sample
-    random = Kernel.rand
+    random = @randomizer.rand
     lhs, rhs = 0, @len - 1
 
     while rhs >= lhs

metadata

@@ -1,8 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: weighted_distribution
 version: !ruby/object:Gem::Version
-  version: 1.0.0
-  prerelease:
+  version: 1.1.0
 platform: ruby
 authors:
 - Jonathan Swenson
@@ -18,13 +17,11 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
-    none: false
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
-    none: false
   prerelease: false
   type: :development
 - !ruby/object:Gem::Dependency
@@ -34,13 +31,11 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
-    none: false
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
-    none: false
   prerelease: false
   type: :development
 description: Sample a weighted distribution of objects.
@@ -53,6 +48,7 @@ files:
 homepage: http://github.com/jonathanswensonsc/weighted_distribution
 licenses:
 - MIT
+metadata: {}
 post_install_message:
 rdoc_options: []
 require_paths:
@@ -62,17 +58,16 @@ required_ruby_version: !ruby/object:Gem::Requirement
   - - '>='
     - !ruby/object:Gem::Version
       version: '0'
-  none: false
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - '>='
     - !ruby/object:Gem::Version
       version: '0'
-  none: false
 requirements: []
 rubyforge_project:
-rubygems_version: 1.8.24
+rubygems_version: 2.1.9
 signing_key:
-specification_version: 3
+specification_version: 4
 summary: Sample a weighted distribution of objects.
 test_files: []
+has_rdoc: