BigW Consortium Gitlab

benchmark_matchers.rb 1.41 KB
Newer Older
1 2 3
module BenchmarkMatchers
  extend RSpec::Matchers::DSL

4 5 6 7
  def self.included(into)
    into.extend(ClassMethods)
  end

8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
  matcher :iterate_per_second do |min_iterations|
    supports_block_expectations

    match do |block|
      @max_stddev ||= 30

      @entry = benchmark(&block)

      expect(@entry.ips).to be >= min_iterations
      expect(@entry.stddev_percentage).to be <= @max_stddev
    end

    chain :with_maximum_stddev do |value|
      @max_stddev = value
    end

    description do
      "run at least #{min_iterations} iterations per second"
    end

    failure_message do
      ips    = @entry.ips.round(2)
      stddev = @entry.stddev_percentage.round(2)

      "expected at least #{min_iterations} iterations per second " \
        "with a maximum stddev of #{@max_stddev}%, instead of " \
        "#{ips} iterations per second with a stddev of #{stddev}%"
    end
  end

  # Benchmarks the given block and returns a Benchmark::IPS::Report::Entry.
  def benchmark(&block)
    report = Benchmark.ips(quiet: true) do |bench|
41 42 43
      bench.report do
        instance_eval(&block)
      end
44 45 46 47
    end

    report.entries[0]
  end
48 49 50 51 52 53 54 55 56 57 58 59 60

  module ClassMethods
    # Wraps around rspec's subject method so you can write:
    #
    #     benchmark_subject { SomeClass.some_method }
    #
    # instead of:
    #
    #     subject { -> { SomeClass.some_method } }
    def benchmark_subject(&block)
      subject { block }
    end
  end
61
end