weighted_list 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 79a123c813e3841e14a5e3c69cd9cabad1a235e8a31ad072d0413b633714a6cf
-  data.tar.gz: 8517d24ed3f7cb6a0f003fe217532b2cf61e01fbc1b6c41b166d475e1cd6f227
+  metadata.gz: 8c25829111b4df390b8c30377bd879709f9e9c680498bc6687feaad4f30c21e4
+  data.tar.gz: 5d6a2c7cd763d2bb4be4c55bbc7c5d5925a108b68dbf16ae1a9ee5b6769fa16b
 SHA512:
-  metadata.gz: e055b1d48cbe320e0aac31a57a31d270fea8b5b92a7c81e3ec15a1a1b3e26654070262dc4e3221819914c3efe245ba9ab0552aaefdb1f0951977c1a4635d7379
-  data.tar.gz: 320b9a7bd0a458983d58e1b74023842524b82c5fd9c35232049c1b17dd99af6b7ee1f3986c3907db4f346e1ce9f776da0752735134201f1dbd106a6d82ad43c6
+  metadata.gz: fbbcd08a9bdfeffdd2779710d85aaa18497a72d3bcd8daf54e1a21cdd5978f2a12bde3d4144d263ad571ddaab1ea7e5ce256a536edd7ee6b6981a28b1781596c
+  data.tar.gz: f82c14b44bb1fbf8954978ec2a39769cb3586c40b24487fecb5c7a43effb5af838688fe0f95e866c1aaf4e29598648add8a71da3c33a42941d3c5e1605a3ab79

data/.gitignore

@@ -8,3 +8,4 @@
 /tmp/
 .rspec_status
 .DS_Store
+Gemfile.lock

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    weighted_list (0.1.0)
+    weighted_list (0.2.0)
 
 GEM
   remote: https://rubygems.org/

data/lib/weighted_list.rb

@@ -1,13 +1,14 @@
 # frozen_string_literal: true
 
+require 'weighted_list/normalizer'
 require 'weighted_list/version'
 require 'weighted_list/sampler'
 
 class WeightedList
   include Enumerable
 
-  def initialize(hash)
-    @hash = hash
+  def initialize(collection)
+    @hash = Normalizer.call(collection)
   end
 
   def each(&block)

data/lib/weighted_list/normalizer.rb

@@ -0,0 +1,25 @@
+class WeightedList
+  class Normalizer
+    def self.call(collection)
+      new(collection).call
+    end
+
+    def initialize(collection)
+      @hash = collection.to_h
+    end
+
+    def call
+      hash.each_with_object({}) do |(item, weight), normalized_hash|
+        normalized_hash[item] = weight.fdiv(total_weight)
+      end
+    end
+
+    private
+
+    attr_reader :hash
+
+    def total_weight
+      @total_weight ||= hash.values.reduce(&:+)
+    end
+  end
+end

data/lib/weighted_list/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class WeightedList
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

data/weighted_list.gemspec

@@ -12,7 +12,7 @@ Gem::Specification.new do |spec|
   spec.email         = ['jayson.virissimo@asu.edu']
 
   spec.summary       = 'Sample from a weighted list.'
-  spec.description   = 'Choose random elements from a weighted list with or without replacement.'
+  spec.description   = 'Sample from an (enumerable) weighted list without replacement.'
   spec.homepage      = 'https://github.com/jaysonvirissimo/weighted_list'
   spec.license       = 'MIT'
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: weighted_list
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Jayson Virissimo
@@ -52,7 +52,7 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.0'
-description: Choose random elements from a weighted list with or without replacement.
+description: Sample from an (enumerable) weighted list without replacement.
 email:
 - jayson.virissimo@asu.edu
 executables: []
@@ -70,6 +70,7 @@ files:
 - bin/console
 - bin/setup
 - lib/weighted_list.rb
+- lib/weighted_list/normalizer.rb
 - lib/weighted_list/sampler.rb
 - lib/weighted_list/version.rb
 - weighted_list.gemspec