From 17278e8ca69d79f4d37e65830c95cc6004c2d8f5 Mon Sep 17 00:00:00 2001
From: Yukihiro Matsumoto <matz@ruby-lang.org>
Date: Fri, 19 Oct 2012 21:16:18 +0900
Subject: add symbol style to Enumerable#inject

---
 mrblib/enum.rb | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'mrblib/enum.rb')

diff --git a/mrblib/enum.rb b/mrblib/enum.rb
index 666ada294..29422716c 100644
--- a/mrblib/enum.rb
+++ b/mrblib/enum.rb
@@ -199,7 +199,12 @@ module Enumerable
   #
   # ISO 15.3.2.2.11
   def inject(*args, &block)
-    raise ArgumentError, "too many arguments" if args.size > 2
+    raise ArgumentError, "too many arguments" if args.size > 3
+    if Symbol === args[-1]
+      sym = args[-1]
+      block = ->(x,y){x.send(sym,y)}
+      args.pop
+    end
     if args.empty?
       flag = true  # no initial argument
       result = nil
@@ -218,6 +223,7 @@ module Enumerable
     }
     result
   end
+  alias reduce inject
 
   ##
   # Alias for collect
-- 
cgit v1.2.3