compare_time 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e43f2fe9a34935b3f0f1a567bb1a1c73c59808b4
-  data.tar.gz: 5ab7f08defcc6467bb816d8797a7b047bd11a79e
+  metadata.gz: 3d0b37a69d25daefd3d2e93996fe4dd6ca60d34e
+  data.tar.gz: ec7ec75ef5192f810673b999e6d6256fa1e0ea33
 SHA512:
-  metadata.gz: 2dff6ce0e5875167147e0a1b2399aa24d76c153563b0b3334e0952a9feeabdf9fb1e8d620c9a087e373064369eea6e3f535266521efedcb5ccd042ebca42d679
-  data.tar.gz: 713590aba2ff1d7b5355b206963815b77ed8d88eba95f5c879ffc3ffd374dca75724a0a29d96780a99ba7932dcfb47d52cf21f6c9b6fdc9142c5ce1306a519fb
+  metadata.gz: c1c65115a4879d3c65d9f74688d2d366df5c92e2cadb9d5aa1206b9a36a244ee7feb6eadb4fc8ed029d8884b3771490ea14cc7168dfeef36c40d1a72908d95ba
+  data.tar.gz: 8c3e94f78ac697ec3110cd8495838c44f72ca4361864ad532381db3eb26ddd64461d3bded760e250f745c8164a5ec085819ec4df573657858bcf09c054bd3afe

data/lib/compare_time.rb

@@ -2,46 +2,68 @@ require 'stringio'
 require 'benchmark'
 require 'colorize'
 
-# CompareTime.new.compare(:whatever_name_you_want) {
-#  ...
-# }.with(:whatever_name_you_want2) {
-#  ...
-# }.print_results
-
 class CompareTime
   attr_reader :benchmarks
 
-  def initialize()
-    # add how many times
+  def initialize(repetitions = 1, silence_output: true)
+    @silence_output = silence_output
+    @repetitions = repetitions
     @benchmarks = {}
   end
 
   def compare(symbol, &block)
-    execute_and_save(symbol, block)
-    self
+    if @silence_output
+      silence_stdout { execute_and_save(symbol, block) }
+    else
+      execute_and_save(symbol, block)
+    end and self
   end
 
-  def results
-    @benchmarks.sort_by(&:last).map do |arr|
-      "#{arr[0]}: #{'%.10f' % arr[1]}"
-    end
+  def sort_results
+    @benchmarks.sort_by(&:last)
   end
 
   def print_results
-    calculated_results = results
-    puts calculated_results[0].colorize(:green)
+    serialized_results = sort_results.map { |res| serialize_result(res) }
+    puts serialized_results[0].colorize(:green)
 
-    calculated_results.drop(1).each do |res|
+    serialized_results.drop(1).each do |res|
       puts res
     end
   end
 
   alias_method :with, :compare
 
-  private def execute_and_save(symbol, block)
+  private
+
+  def serialize_result(arr)
+    "#{arr[0]}: #{'%.10f' % arr[1]}"
+  end
+
+  def execute_and_save(symbol, block)
+    if @repetitions == 1
+      @benchmarks[symbol] = single_repetition(block)
+    else
+      @benchmarks[symbol] = multiple_repetitions(block)
+    end
+  end
+
+  def single_repetition(block)
+    Benchmark.realtime(&block)
+  end
+
+  def multiple_repetitions(block)
+    arr = []
+    @repetitions.times do
+      arr << single_repetition(block)
+    end
+    arr.inject(0.0) { |sum, el| sum + el } / arr.size
+  end
+
+  def silence_stdout(&block)
     original_stdout = $stdout
     $stdout = StringIO.new
-    @benchmarks[symbol] = Benchmark.realtime(&block)
+    block.call
   ensure
     $stdout = original_stdout
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: compare_time
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - Bartek Gladecki