summaryrefslogtreecommitdiff
path: root/spec/mspec/lib/mspec/matchers/infinity.rb
blob: 0949fd47ebae9c14b13fd1394344aa49bf9bc840 (plain)
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
class InfinityMatcher
  def initialize(expected_sign)
    @expected_sign = expected_sign
  end

  def matches?(actual)
    @actual = actual
    @actual.kind_of?(Float) && @actual.infinite? == @expected_sign
  end

  def failure_message
    ["Expected #{@actual}", "to be #{"-" if @expected_sign == -1}Infinity"]
  end

  def negative_failure_message
    ["Expected #{@actual}", "not to be #{"-" if @expected_sign == -1}Infinity"]
  end
end

class Object
  def be_positive_infinity
    InfinityMatcher.new(1)
  end

  def be_negative_infinity
    InfinityMatcher.new(-1)
  end
end