From 44763ebc749518693ae85b59c8e22cfe8ba8819f Mon Sep 17 00:00:00 2001 From: Jun Hiroe Date: Fri, 21 Mar 2014 11:50:09 +0900 Subject: Add Enumerable#{flat_map,collect_concat} --- mrbgems/mruby-enum-ext/test/enum.rb | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'mrbgems/mruby-enum-ext/test/enum.rb') diff --git a/mrbgems/mruby-enum-ext/test/enum.rb b/mrbgems/mruby-enum-ext/test/enum.rb index 065ef7f5f..9523b497e 100644 --- a/mrbgems/mruby-enum-ext/test/enum.rb +++ b/mrbgems/mruby-enum-ext/test/enum.rb @@ -60,3 +60,9 @@ assert("Enumerable#count") do assert_equal 2, a.count(2) assert_equal 3, a.count{|x| x % 2 == 0} end + +assert("Enumerable#flat_map") do + assert_equal [1, 2, 3, 4], [1, 2, 3, 4].flat_map { |e| e } + assert_equal [1, -1, 2, -2, 3, -3, 4, -4], [1, 2, 3, 4].flat_map { |e| [e, -e] } + assert_equal [1, 2, 100, 3, 4, 100], [[1, 2], [3, 4]].flat_map { |e| e + [100] } +end -- cgit v1.2.3