From 02ac5dd9f1bdf0b2c99a9a86d3f1a029ad09a1c0 Mon Sep 17 00:00:00 2001 From: Jun Hiroe Date: Tue, 18 Mar 2014 22:48:57 +0900 Subject: Add Enumerable#count --- mrbgems/mruby-enum-ext/mrblib/enum.rb | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'mrbgems/mruby-enum-ext') diff --git a/mrbgems/mruby-enum-ext/mrblib/enum.rb b/mrbgems/mruby-enum-ext/mrblib/enum.rb index 90f321596..1ca62b673 100644 --- a/mrbgems/mruby-enum-ext/mrblib/enum.rb +++ b/mrbgems/mruby-enum-ext/mrblib/enum.rb @@ -208,4 +208,22 @@ module Enumerable a end end + + def count(v=nil, &block) + count = 0 + if block + self.each do |e| + count += 1 if block.call(e) + end + else + if v == nil + self.each { count += 1 } + else + self.each do |e| + count += 1 if e == v + end + end + end + count + end end -- cgit v1.2.3 From 17fdd815731490bd3d68233d6749a908646abbf2 Mon Sep 17 00:00:00 2001 From: Jun Hiroe Date: Tue, 18 Mar 2014 23:02:34 +0900 Subject: Add Enumerable#count test --- mrbgems/mruby-enum-ext/test/enum.rb | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'mrbgems/mruby-enum-ext') diff --git a/mrbgems/mruby-enum-ext/test/enum.rb b/mrbgems/mruby-enum-ext/test/enum.rb index a14d7f492..065ef7f5f 100644 --- a/mrbgems/mruby-enum-ext/test/enum.rb +++ b/mrbgems/mruby-enum-ext/test/enum.rb @@ -53,3 +53,10 @@ assert("Enumerable#first") do assert_equal [1, 2, 3], a.first(10) assert_nil [].first end + +assert("Enumerable#count") do + a = [1, 2, 4, 2] + assert_equal 4, a.count + assert_equal 2, a.count(2) + assert_equal 3, a.count{|x| x % 2 == 0} +end -- cgit v1.2.3