Add support for :roughly parameter on Integer matching

lib/machete/matchers/integer_matcher.ex

@@ -11,7 +11,8 @@ defmodule Machete.IntegerMatcher do
             strictly_negative: nil,
             nonzero: nil,
             min: nil,
-            max: nil
+            max: nil,
+            roughly: nil
 
   @typedoc """
   Describes an instance of this matcher
@@ -28,7 +29,8 @@ defmodule Machete.IntegerMatcher do
           {:strictly_negative, boolean()},
           {:nonzero, boolean()},
           {:min, integer()},
-          {:max, integer()}
+          {:max, integer()},
+          {:roughly, integer()}
         ]
 
   @doc """
@@ -43,6 +45,7 @@ defmodule Machete.IntegerMatcher do
   * `nonzero`: When `true`, requires the matched integer be nonzero
   * `min`: Requires the matched integer be greater than or equal to the specified value
   * `max`: Requires the matched integer be less than or equal to the specified value
+  * `roughly`: Requires the matched integer be within 5% of the specified value. Raises if set to `0`
 
   Examples:
 
@@ -108,6 +111,12 @@ defmodule Machete.IntegerMatcher do
 
       iex> assert 2 ~> integer(max: 2)
       true
+
+      iex> assert 95 ~> integer(roughly: 100)
+      true
+
+      iex> refute 94 ~> integer(roughly: 100)
+      false
   """
   @spec integer(opts()) :: t()
   def integer(opts \\ []), do: struct!(__MODULE__, opts)
@@ -121,7 +130,8 @@ defmodule Machete.IntegerMatcher do
            nil <- matches_strictly_negative(b, a.strictly_negative),
            nil <- matches_nonzero(b, a.nonzero),
            nil <- matches_min(b, a.min),
-           nil <- matches_max(b, a.max) do
+           nil <- matches_max(b, a.max),
+           nil <- matches_roughly(b, a.roughly) do
       end
     end
 
@@ -167,5 +177,13 @@ defmodule Machete.IntegerMatcher do
       do: mismatch("#{safe_inspect(b)} is greater than #{max}")
 
     defp matches_max(_, _), do: nil
+
+    defp matches_roughly(_b, 0), do: raise("`roughly` parameter cannot be 0")
+
+    defp matches_roughly(b, roughly)
+         when is_integer(roughly) and (b / roughly < 0.95 or b / roughly > 1.05),
+         do: mismatch("#{safe_inspect(b)} is more than 5% different than #{roughly}")
+
+    defp matches_roughly(_, _), do: nil
   end
 end

test/machete/matchers/integer_matcher_test.exs

@@ -57,4 +57,14 @@ defmodule IntegerMatcherTest do
   test "produces a useful mismatch for max mismatch" do
     assert 2 ~>> integer(max: 1) ~> mismatch("2 is greater than 1")
   end
+
+  test "produces a useful mismatch for roughly mismatch" do
+    assert 94 ~>> integer(roughly: 100) ~> mismatch("94 is more than 5% different than 100")
+  end
+
+  test "raises when provided with a roughly value of 0" do
+    assert_raise(RuntimeError, "`roughly` parameter cannot be 0", fn ->
+      assert 94 ~>> integer(roughly: 0)
+    end)
+  end
 end