diff options
author | Yusuke Endoh <mame@ruby-lang.org> | 2021-04-21 13:02:29 +0900 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-04-21 13:02:29 +0900 |
commit | fb04c69418ceee696a114fe31279cf3a5ea16d30 (patch) | |
tree | 0f8d78e40d15172c80ec0853fb37ea755049394c | |
parent | d427e3cd6f1b39624c9935ade25b4dcb334bee36 (diff) |
array.c (rb_ary_zip): take only as many as needed from an Enumerator (#4389)
[Bug #17814]
Notes
Notes:
Merged-By: mame <mame@ruby-lang.org>
-rw-r--r-- | array.c | 4 | ||||
-rw-r--r-- | test/ruby/test_array.rb | 11 |
2 files changed, 13 insertions, 2 deletions
@@ -4311,10 +4311,9 @@ static VALUE take_i(RB_BLOCK_CALL_FUNC_ARGLIST(val, cbarg)) { VALUE *args = (VALUE *)cbarg; - if (args[1] == 0) rb_iter_break(); - else args[1]--; if (argc > 1) val = rb_ary_new4(argc, argv); rb_ary_push(args[0], val); + if (--args[1] == 0) rb_iter_break(); return Qnil; } @@ -4324,6 +4323,7 @@ take_items(VALUE obj, long n) VALUE result = rb_check_array_type(obj); VALUE args[2]; + if (n == 0) return result; if (!NIL_P(result)) return rb_ary_subseq(result, 0, n); result = rb_ary_new2(n); args[0] = result; args[1] = (VALUE)n; diff --git a/test/ruby/test_array.rb b/test/ruby/test_array.rb index bf6a728574..4a9863cc54 100644 --- a/test/ruby/test_array.rb +++ b/test/ruby/test_array.rb @@ -2735,6 +2735,17 @@ class TestArray < Test::Unit::TestCase assert_equal [[42, 1]], [42].zip(r), bug8153 end + def test_zip_with_enumerator + bug17814 = "ruby-core:103513" + + step = 0.step + e = Enumerator.produce { step.next } + a = %w(a b c) + assert_equal([["a", 0], ["b", 1], ["c", 2]], a.zip(e)) + assert_equal([["a", 3], ["b", 4], ["c", 5]], a.zip(e)) + assert_equal([["a", 6], ["b", 7], ["c", 8]], a.zip(e)) + end + def test_transpose assert_equal([[1, :a], [2, :b], [3, :c]], [[1, 2, 3], [:a, :b, :c]].transpose) |