example update

Author: v0dro

examples/flat_vs_flat_map.rb

@@ -1,12 +1,26 @@
+require 'benchmark/plot'
+
+class TestArray
+  attr_reader :arr
+
+  def initialize arr
+    @arr = arr
+  end
+
+  def to_s
+    @arr.size.to_s
+  end
+end
+
 test_data = [5, 25, 50, 75, 100, 125, 150, 175, 200,250,300]
-test_data.map! {|e| Array.new(e) {|i| i} }
+test_data.map! {|e| TestArray.new(Array.new(e) {|i| i}) }
 
 Benchmark.plot(test_data) do |x|
   x.report("map.flatten") do |data|
-    data.map { [nil]  }.flatten
+    data.arr.map { [nil]  }.flatten
   end
 
   x.report("flat_map") do |data|
-    data.flat_map { [nil] }
+    data.arr.flat_map { [nil] }
   end
-end
+end

examples/mapflat_vs_flat_map/flat_vs_flat_map.rb

@@ -1,14 +0,0 @@
-require 'nmatrix'
-require 'matrix'
-
-test_data = [5,10,15,20,25,30,35]
-
-Benchmark.plot(test_data) do |x, data|
-  x.report("nmatrix creation") do
-    Array.new(data) { |i|  i }
-  end
-
-  x.report("matrix creation") do
-    Array.new(data) { |i|  i }
-  end
-end