Add block mode to assert_equal

1 files changed, 10 insertions, 1 deletions

diff --git a/test/assert.rb b/test/assert.rb
index 6d727e889..f7c2c97b9 100644
--- a/test/assert.rb
+++ b/test/assert.rb
@@ -99,7 +99,16 @@ def assert_false(ret, msg = nil, diff = nil)
   !ret
 end
 
-def assert_equal(exp, act, msg = nil)
+def assert_equal(arg1, arg2 = nil, arg3 = nil)
+  if block_given?
+    exp = yield
+    act = arg1
+    msg = arg2
+  else
+    exp = arg1
+    act = arg2
+    msg = arg3
+  end
   msg = "Expected to be equal" unless msg
   diff = assertion_diff(exp, act)
   assert_true(exp == act, msg, diff)