From 9326c514c505dd47d2f25505c294d26b32d12336 Mon Sep 17 00:00:00 2001 From: "Yukihiro \"Matz\" Matsumoto" Date: Sat, 20 Jul 2024 14:20:15 +0900 Subject: [PATCH] mruby-enum-ext: add Enumerable#each_entry test --- mrbgems/mruby-enum-ext/test/enum.rb | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/mrbgems/mruby-enum-ext/test/enum.rb b/mrbgems/mruby-enum-ext/test/enum.rb index 98ace670e..f0441ba15 100644 --- a/mrbgems/mruby-enum-ext/test/enum.rb +++ b/mrbgems/mruby-enum-ext/test/enum.rb @@ -202,3 +202,22 @@ assert("Enumerable#grep_v") do assert_equal [1, 2, 3, 4, 5, 0], a.grep_v(6..8) assert_equal [2, 4, 6, 8, 10], a.grep_v(0) {|v| v * 2} end + +assert("Enumerable#each_entry") do + each_entry_test = Class.new { + include Enumerable + def each + yield 1 + yield 1, 2 + yield + end + } + e = each_entry_test.new + a = [] + e.each_entry {|v| + a.push(v) + } + assert_equal 1, a[0] + assert_equal [1,2], a[1] + assert_equal nil, a[2] +end