randomfromrange 0.0.2 → 0.0.3

data/lib/randomfromrange.rb

@@ -5,10 +5,14 @@ Kernel.class_eval do
 		if(v.respond_to? :to_i)
 			return old_rand(v.to_i.abs)
 		elsif(v.respond_to? :begin and v.respond_to? :end)
-		  b = (v.begin.to_s =~ /^[0-9]+\.([0-9]+)$/) ? $1.to_s : ''
-		  e = (v.end.to_s =~ /^[0-9]+\.([0-9]+)$/) ? $1.to_s : ''
-		  precision = [b.length, e.length].max
-			return ((v.end-v.begin)*old_rand() + v.begin).round(precision)
+		  first = nil
+		  v.step(1) { |n| first = n }
+		  precision = ((first.to_s =~ /^[0-9]+\.([0-9]+)$/) ? $1.to_s : '').length
+		  if precision > 0
+			  return (('%.' + precision.to_s + 'f') % ((v.end-v.begin)*old_rand() + v.begin)).to_f
+			else
+			  return ((v.end-v.begin)*old_rand() + v.begin).round
+			end
 		else
 			raise TypeError, 'can\'t convert ' + v.class + ' into Integer'
 		end	

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: randomfromrange
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors: