random-walk 0.0.1 → 0.0.2

data/README.rdoc

@@ -2,18 +2,25 @@
 
 This is an extremely simple gem for generating random-walk test data in an array. It's useful for statistics testing, where random data falls short for testing something like simple regression.
 
-The resulting array will always stay within the limits provided (inclusive). The random-walk adds +1, 0, or -1 on each iteration.
+The resulting array will always stay within the range provided (inclusive). The random-walk adds +1, 0, or -1 on each iteration.
 
-## Usage
+== Usage
 
-```ruby
-require 'random-walk'
+    require 'random-walk'
 
-# RandomWalk.generate(limits, array_length)
-RandomWalk.generate(10..40, 5) # [11, 12, 11, 10, 10]
-RandomWalk.generate(10..40, 5) # [27, 28, 29, 30, 29]
-RandomWalk.generate(10..40, 5) # [27, 28, 29, 30, 29]
-```
+    # RandomWalk.generate(range, array_length)
+    RandomWalk.generate(10..40, 5) # [11, 12, 11, 10, 10]
+    RandomWalk.generate(10..40, 5) # [27, 28, 29, 30, 29]
+
+    # generate_vectors(x_range, y_range, vector_length)
+    rw = RandomWalk.new
+    rw.generate_vectors(0..99, 0..1023, 5)
+    rw.x # [78, 79, 80, 81, 81]
+    rw.y # [700, 701, 702, 702, 701]
+
+    # fit_y_to_x(fit_chance)
+    rw.fit_y_to_x(100) # Perfect fit. If x goes up by 1, y goes up by 1
+    rw.fit_y_to_x(50) # 50% chance. If x goes up by 1, y has a 50 percent chance of going up by 1
 
 == Contributing to random-walk
  

data/VERSION

@@ -1 +1 @@
-0.0.1
+0.0.2

data/lib/random-walk.rb

@@ -1,8 +1,9 @@
 class RandomWalk
+  attr_accessor :x, :y
+
   def self.generate(limits, array_length)
     starting_point = Random.rand(limits)
     output_array = [starting_point]
-
     (array_length - 1).times do
       add_or_subtract = Random.rand(-1..1)
       end_point = output_array.last + add_or_subtract
@@ -12,7 +13,24 @@ class RandomWalk
         output_array << output_array.last
       end
     end
-
     return output_array
   end
+
+  def generate_vectors(x_limits, y_limits, vector_length)
+    self.x = RandomWalk.generate(x_limits, vector_length)
+    self.y = RandomWalk.generate(y_limits, vector_length)
+  end
+
+  def fit_y_to_x(fit_chance)
+    raise 'Fit chance must be between 0 and 100' unless fit_chance <= 100 && fit_chance >= 0
+    self.x.each_with_index do |value, i|
+      if self.x[i] && self.x[i+1]
+        x_diff = self.x[i] - self.x[i+1]
+        rand = Random.rand(100 / fit_chance)
+        if rand == 0
+          self.y[i+1] = self.y[i] - x_diff
+        end
+      end
+    end
+  end
 end

data/random-walk.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "random-walk"
-  s.version = "0.0.1"
+  s.version = "0.0.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Evan Tann"]
-  s.date = "2013-01-06"
+  s.date = "2013-01-07"
   s.description = "Generates an array of random-walk data within user-specified limits and of an arbitrary length for use in testing."
   s.email = "egtann@gmail.com"
   s.extra_rdoc_files = [

data/spec/random-walk_spec.rb

@@ -1,17 +1,62 @@
 require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 
 describe "RandomWalk" do
-  it "returns results between given limits" do
-    5.times do
-      results = RandomWalk.generate(0..10, 1000)
-      results.each do |result|
-        (result.should be >= 0) && (result.should be <= 10)
+  describe "generate" do
+    it "returns results between given limits" do
+      5.times do
+        results = RandomWalk.generate(0..10, 1000)
+        results.each do |result|
+          result.should be >= 0
+          result.should be <= 10
+        end
       end
     end
+
+    it "returns an array of the correct length" do
+      results = RandomWalk.generate(100..200, 250)
+      results.length.should == 250
+    end
+  end
+
+  describe "generate_vectors" do
+    let(:rw) { RandomWalk.new }
+
+    it "returns results between given limits" do
+      5.times do
+        rw.generate_vectors(0..10, 0..10, 1000)
+        rw.x.each do |result|
+          result.should be >= 0
+          result.should be <= 10
+        end
+        rw.y.each do |result|
+          result.should be >= 0
+          result.should be <= 10
+        end
+      end
+    end
+
+    it "returns 2 arrays of the correct length" do
+      rw.generate_vectors(0..99, 0..1023, 500)
+      rw.x.length.should == 500
+      rw.y.length.should == 500
+    end
   end
 
-  it "returns an array of the correct length" do
-    results = RandomWalk.generate(100..200, 250)
-    results.length.should == 250
+  describe "fit_y_to_x" do
+    let(:rw) { RandomWalk.new }
+
+    it "fits the y array to the x array perfectly" do
+      rw.generate_vectors(0..99, 250..1000, 500)
+      fit_chance = 100
+      rw.fit_y_to_x(fit_chance)
+      diff_x, diff_y = 0, 0
+      rw.x.each_with_index do |result, i|
+        if rw.x[i+1]
+          diff_x += rw.x[i] - rw.x[i+1]
+          diff_y += rw.y[i] - rw.y[i+1]
+        end
+      end
+      diff_x.should == diff_y
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: random-walk
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-01-06 00:00:00.000000000 Z
+date: 2013-01-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -111,7 +111,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -3124899964672894423
+      hash: -3032803507122637849
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: