lib/akero/benchmark.rb in akero-1.0.4 vs lib/akero/benchmark.rb in akero-1.1.0

@@ -1,5 +1,6 @@
+# frozen_string_literal: true
 require 'akero'
 require 'gnuplot'
 require 'b'
 
 class Akero
@@ -10,14 +11,14 @@
         b_size
         b_timing
       end
 
       def b_size
-        puts "Running size benchmark..."
+        puts 'Running size benchmark...'
 
         rnd = Random.new
-        msg_sizes = (8..14).map{|x| 2**x}
+        msg_sizes = (8..14).map { |x| 2**x }
         modes = [[2048, false], [2048, true]]
         results = {}
         modes.each do |mode|
           alice = Akero.new(mode[0])
           bob   = Akero.new(mode[0])
@@ -30,74 +31,74 @@
 
         plot('benchmark/bm_size.png', results, 'Message size overhead', 'Input size (bytes)', 'x')
       end
 
       def b_timing
-        puts "Running timing benchmark..."
+        puts 'Running timing benchmark...'
 
-        msg_sizes = (8..18).map{|x| 2**x}
+        msg_sizes = (8..18).map { |x| 2**x }
         modes = [[2048, false], [2048, true]]
 
         rnd = Random.new
 
         rounds = 50
         results = []
         modes.each do |mode|
-          results << B.enchmark("ENCRYPT #{mode[0]} bits, ascii_armor=#{mode[1]?1:0}", :rounds => rounds, :compare => :mean) do
+          results << B.enchmark("ENCRYPT #{mode[0]} bits, ascii_armor=#{mode[1] ? 1 : 0}", rounds: rounds, compare: :mean) do
             alice = Akero.new(mode[0])
             bob   = Akero.new(mode[0])
-            msg_sizes.each_with_index do |msize, i|
+            msg_sizes.each_with_index do |msize, _i|
               msg = rnd.bytes(msize)
               job "msg_size #{msize}" do
                 alice.encrypt(bob.public_key, msg, mode[1])
               end
             end
           end
         end
 
         modes.each do |mode|
-          results << B.enchmark("DECRYPT #{mode[0]} bits, ascii_armor=#{mode[1]?1:0}", :rounds => rounds, :compare => :mean) do
+          results << B.enchmark("DECRYPT #{mode[0]} bits, ascii_armor=#{mode[1] ? 1 : 0}", rounds: rounds, compare: :mean) do
             alice = Akero.new(mode[0])
             bob   = Akero.new(mode[0])
-            msg_sizes.each_with_index do |msize, i|
+            msg_sizes.each_with_index do |msize, _i|
               msg = rnd.bytes(msize)
               msg = alice.encrypt(bob.public_key, msg, mode[1])
               job "msg_size #{msize}" do
                 bob.receive(msg)
               end
             end
           end
         end
 
         modes.each do |mode|
-          results << B.enchmark("SIGN #{mode[0]} bits, ascii_armor=#{mode[1]?1:0}", :rounds => rounds, :compare => :mean) do
+          results << B.enchmark("SIGN #{mode[0]} bits, ascii_armor=#{mode[1] ? 1 : 0}", rounds: rounds, compare: :mean) do
             alice = Akero.new(mode[0])
             bob   = Akero.new(mode[0])
-            msg_sizes.each_with_index do |msize, i|
+            msg_sizes.each_with_index do |msize, _i|
               msg = rnd.bytes(msize)
               job "msg_size #{msize}" do
                 alice.sign(msg, mode[1])
               end
             end
           end
         end
 
         ds = {}
         title = []
-        results.each_with_index do |r, i|
-          r.each_with_index do |v,j|
-            (ds[v[:group]] ||= []) << [msg_sizes[j],v[:rate]]
+        results.each_with_index do |r, _i|
+          r.each_with_index do |v, j|
+            (ds[v[:group]] ||= []) << [msg_sizes[j], v[:rate]]
           end
         end
 
-        plot('benchmark/bm_rate.png', ds, "Throughput", 'Input size (bytes)', 'Messages/sec')
+        plot('benchmark/bm_rate.png', ds, 'Throughput', 'Input size (bytes)', 'Messages/sec')
       end
 
       def plot(path, ds, label, xlabel, ylabel)
         Gnuplot.open do |gp|
-          Gnuplot::Plot.new( gp ) do |plot|
-            plot.terminal "png"
+          Gnuplot::Plot.new(gp) do |plot|
+            plot.terminal 'png'
             plot.output path
             plot.title  label
             plot.xlabel xlabel
             plot.ylabel ylabel
             plot.xtics :axis
@@ -106,16 +107,16 @@
             plot.style 'line 11 lc rgb "#808080" lt 1'
             plot.set 'border 3 back ls 11'
             plot.set 'tics nomirror'
             plot.style 'line 12 lc rgb "#808080" lt 0 lw 1'
             plot.set 'grid back ls 12'
-            i=0
+            i = 0
             ds.each do |k, v|
-              plot.data << Gnuplot::DataSet.new([v.collect {|x| x[0]}, v.collect {|x| x[1]}]) do |_ds|
-                _ds.with = "lp ls #{i+1}"
+              plot.data << Gnuplot::DataSet.new([v.collect { |x| x[0] }, v.collect { |x| x[1] }]) do |_ds|
+                _ds.with = "lp ls #{i + 1}"
                 _ds.title = k
               end
-              i+=1
+              i += 1
             end
           end
         end
       end
     end