From 821f6d42adf92c09a0df082c14c2c4ade21f8c78 Mon Sep 17 00:00:00 2001
From: Daniel Bovensiepen <daniel@bovensiepen.net>
Date: Fri, 14 Jun 2013 21:32:13 +0800
Subject: Add block mode to assert_equal

---
 test/assert.rb | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

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)
-- 
cgit v1.2.3