benchmark.rb

require "benchmark"

$count = 5
$item_count = 100_000
$iterations = 1_000


def setup
    (1..$count).map{|x| {:num => x, :data => (1..$item_count).to_a}}
end


def test_one
    seq = setup()
    Benchmark.measure do
        (1..$iterations).each do
            res = seq.flat_map{|x| x[:data]}
            raise Exception if res.length != $count * $item_count
        end
    end
end


def test_two
    seq = setup()
    Benchmark.measure do
        (1..$iterations).each do
            res = []
            seq.each{|x|  res += x[:data]}
            raise Exception if res.length != $count * $item_count
        end
    end
end


puts "Ran test one in #{test_one}s"
puts "Ran test two in #{test_two}s"