chore(benchmark): dry up benchmark codes

tonytonyjan · Oct 3, 2017 · 5ebd36d · 5ebd36d
1 parent 7d590dc
commit 5ebd36d
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 16 deletions.
diff --git a/benchmark/env.rb b/benchmark/env.rb
@@ -4,6 +4,19 @@
 require 'bundler'
 Bundler.setup(:benchmark)
 
+SAMPLES = {
+  ascii: [
+    %w[al al], %w[martha marhta], %w[jones johnson], %w[abcvwxyz cabvwxyz],
+    %w[dwayne duane], %w[dixon dicksonx], %w[fvie ten]
+  ].freeze,
+  utf8: [
+    %w[馬英九 馬英丸], %w[蔡英文 蔡中文], %w[簡煒航 簡偉航], %w[焦玟綾 焦紋綾],
+    %w[眼球中央電視台 眼球中英電視台], %w[床前明月光 床前日月光],
+    %w[海水退了就知道誰沒穿褲子 海水退了就知道誰沒穿襪子],
+    %w[阿里山的姑娘美如水 阿里山的姑娘沒乳水]
+  ].freeze
+}.freeze
+
 def gem_name_with_version(gem)
   "#{gem} (#{Gem.loaded_specs[gem].version})"
 end
diff --git a/benchmark/native.rb b/benchmark/native.rb
@@ -7,28 +7,23 @@
 require 'hotwater'
 require 'amatch'
 
-ary = [
-  %w[al al], %w[martha marhta], %w[jones johnson], %w[abcvwxyz cabvwxyz],
-  %w[dwayne duane], %w[dixon dicksonx], %w[fvie ten]
-]
-
 n = 100_000
 
 Benchmark.bmbm do |x|
   x.report "jaro_winkler (#{`git rev-parse --short HEAD`.chop!})" do
-    n.times { ary.each { |str1, str2| JaroWinkler.distance(str1, str2) } }
+    n.times { SAMPLES[:ascii].each { |str1, str2| JaroWinkler.distance(str1, str2) } }
   end
 
   x.report gem_name_with_version('fuzzy-string-match') do
     jarow = FuzzyStringMatch::JaroWinkler.create(:native)
-    n.times { ary.each { |str1, str2| jarow.getDistance(str1, str2) } }
+    n.times { SAMPLES[:ascii].each { |str1, str2| jarow.getDistance(str1, str2) } }
   end
 
   x.report gem_name_with_version('hotwater') do
-    n.times { ary.each { |str1, str2| Hotwater.jaro_winkler_distance(str1, str2) } }
+    n.times { SAMPLES[:ascii].each { |str1, str2| Hotwater.jaro_winkler_distance(str1, str2) } }
   end
 
   x.report gem_name_with_version('amatch') do
-    n.times { ary.each { |str1, str2| Amatch::Jaro.new(str1).match(str2) } }
+    n.times { SAMPLES[:ascii].each { |str1, str2| Amatch::Jaro.new(str1).match(str2) } }
   end
 end
diff --git a/benchmark/pure.rb b/benchmark/pure.rb
@@ -5,20 +5,15 @@
 require 'jaro_winkler/jaro_winkler_pure'
 require 'fuzzystringmatch'
 
-ary = [
-  %w[al al], %w[martha marhta], %w[jones johnson], %w[abcvwxyz cabvwxyz],
-  %w[dwayne duane], %w[dixon dicksonx], %w[fvie ten]
-]
-
 n = 10_000
 
 Benchmark.bmbm do |x|
   x.report "jaro_winkler (#{`git rev-parse --short HEAD`.chop!})" do
-    n.times { ary.each { |str1, str2| JaroWinkler.distance(str1, str2) } }
+    n.times { SAMPLES[:ascii].each { |str1, str2| JaroWinkler.distance(str1, str2) } }
   end
 
   x.report gem_name_with_version('fuzzy-string-match') do
     jarow = FuzzyStringMatch::JaroWinkler.create(:pure)
-    n.times { ary.each { |str1, str2| jarow.getDistance(str1, str2) } }
+    n.times { SAMPLES[:ascii].each { |str1, str2| jarow.getDistance(str1, str2) } }
   end
 end