Update benchmark script

Show elapsed time of the last execution.
koichiro · Mar 27, 2012 · b784697 · b784697
1 parent 27166a6
commit b784697
Showing 1 changed file with 13 additions and 14 deletions.
diff --git a/bench/bench.rb b/bench/bench.rb
@@ -1,18 +1,15 @@
+# encoding: utf-8
 require_relative "../lib/rbtree_map"
 
 file = ARGV.shift or raise
 
 def benchmark(&block)
-  begin
+  start = Time.now
-    start = Time.now
+  yield
-    yield
+  (Time.now - start) *1000
-    finished = Time.now
-  ensure
-    puts "$B7P2a;~4V(B #{(finished - start) * 1000} [msec]"
-  end
 end
 
-print "$BF0:n3NG'(B..."
+print "動作確認..."
 map = RBTreeMap.newInstance
 File.open(file).each_line do |line|
   key, value, height = line.chomp.split(',', 3)
@@ -21,15 +18,16 @@ def benchmark(&block)
 File.open(file).each_line do |line|
   key, value, _ = line.split(',', 3)
   if map.get(key) != value
-    raise "$BCM$NIT0lCW(B #{map.get(key)} != #{value} at line #{line}"
+    raise "値の不一致 #{map.get(key)} != #{value} at line #{line}"
   end
 end
 puts "OK"
 
-puts "$B%Y%s%A%^!<%/(B..."
+times = 5
-5.times do |idx|
+puts "ウォームアップ... (#{times} 回)"
-  print "$B%Y%s%A%^!<%/(B #{idx + 1} $B2sL\(B: "
+(times + 1).times do |idx|
-  benchmark do
+  puts "ベンチマーク: " if idx == times
+  elapsed = benchmark {
     map = RBTreeMap.newInstance
     File.open(file).each_line do |line|
       key, value, height = line.chomp.split(',', 3)
@@ -41,5 +39,6 @@ def benchmark(&block)
         puts "wrong value! #{map.get(key)} != #{value}"
       end
     end
-  end
+  }
+  puts elapsed
 end