File: invariants.rb

package info (click to toggle)
ruby-contracts 0.17-2
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 624 kB
  • sloc: ruby: 3,805; makefile: 4; sh: 2
file content (91 lines) | stat: -rw-r--r-- 1,940 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
require "./lib/contracts"
require "benchmark"
require "rubygems"
require "method_profiler"
require "ruby-prof"

class Obj
  include Contracts

  attr_accessor :value
  def initialize value
    @value = value
  end

  Contract Num, Num => Num
  def contracts_add a, b
    a + b
  end
end

class ObjWithInvariants
  include Contracts
  include Contracts::Invariants

  invariant(:value_not_nil) { value != nil }
  invariant(:value_not_string) { !value.is_a?(String) }

  attr_accessor :value
  def initialize value
    @value = value
  end

  Contract Num, Num => Num
  def contracts_add a, b
    a + b
  end
end

def benchmark
  obj = Obj.new(3)
  obj_with_invariants = ObjWithInvariants.new(3)

  Benchmark.bm 30 do |x|
    x.report "testing contracts add" do
      1_000_000.times do |_|
        obj.contracts_add(rand(1000), rand(1000))
      end
    end
    x.report "testing contracts add with invariants" do
      1_000_000.times do |_|
        obj_with_invariants.contracts_add(rand(1000), rand(1000))
      end
    end
  end
end

def profile
  obj_with_invariants = ObjWithInvariants.new(3)

  profilers = []
  profilers << MethodProfiler.observe(Contract)
  profilers << MethodProfiler.observe(Object)
  profilers << MethodProfiler.observe(Contracts::Support)
  profilers << MethodProfiler.observe(Contracts::Invariants)
  profilers << MethodProfiler.observe(Contracts::Invariants::InvariantExtension)
  profilers << MethodProfiler.observe(UnboundMethod)

  10_000.times do |_|
    obj_with_invariants.contracts_add(rand(1000), rand(1000))
  end

  profilers.each { |p| puts p.report }
end

def ruby_prof
  RubyProf.start

  obj_with_invariants = ObjWithInvariants.new(3)

  100_000.times do |_|
    obj_with_invariants.contracts_add(rand(1000), rand(1000))
  end

  result = RubyProf.stop
  printer = RubyProf::FlatPrinter.new(result)
  printer.print(STDOUT)
end

benchmark
profile
ruby_prof if ENV["FULL_BENCH"] # takes some time