diff options
author | yugui <yugui@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2009-01-28 09:20:03 +0000 |
---|---|---|
committer | yugui <yugui@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2009-01-28 09:20:03 +0000 |
commit | 1011775a395e901aa897301dc5fecb9270fc1646 (patch) | |
tree | b05880a19cf90ba3fe5790f9cd9e5d4b55059013 | |
parent | a40ca8c047d8c174d1757dcbc82ec16185144717 (diff) |
introduces r21693 again and merges r21713
* array.c (take_items), enum.c (enum_zip): tries to convert to
array first. [ruby-core:21442]
--
* array.c (take_items): to_ary() raises ArgumentError if cannot to
convert to Array. [ruby-dev:37797]
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_1_9_1@21814 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
-rw-r--r-- | ChangeLog | 9 | ||||
-rw-r--r-- | array.c | 4 | ||||
-rw-r--r-- | enum.c | 8 | ||||
-rw-r--r-- | test/ruby/test_array.rb | 8 | ||||
-rw-r--r-- | test/ruby/test_enum.rb | 8 |
5 files changed, 34 insertions, 3 deletions
@@ -1,3 +1,12 @@ +Wed Jan 21 14:41:48 2009 NAKAMURA Usaku <usa@ruby-lang.org> + + * array.c (take_items): to_ary() raises ArgumentError if cannot to + convert to Array. [ruby-dev:37797] + +Wed Jan 21 21:12:47 2009 Yuki Sonoda (Yugui) <yugui@yugui.jp> + + * introduces r21693 again for merging r21713. + Tue Jan 20 19:12:18 2009 Yuki Sonoda (Yugui) <yugui@yugui.jp> * reverts r21693. @@ -2209,9 +2209,11 @@ take_i(VALUE val, VALUE *args, int argc, VALUE *argv) static VALUE take_items(VALUE obj, long n) { - VALUE result = rb_ary_new2(n); + VALUE result = rb_check_array_type(obj); VALUE args[2]; + if (!NIL_P(result)) return rb_ary_subseq(result, 0, n); + result = rb_ary_new2(n); args[0] = result; args[1] = (VALUE)n; rb_block_call(obj, rb_intern("each"), 0, 0, take_i, (VALUE)args); return result; @@ -1558,13 +1558,17 @@ enum_zip(int argc, VALUE *argv, VALUE obj) ID conv; NODE *memo; VALUE result = Qnil; + VALUE args = rb_ary_new4(argc, argv); int allary = Qtrue; + argv = RARRAY_PTR(args); for (i=0; i<argc; i++) { - if (TYPE(argv[i]) != T_ARRAY) { + VALUE ary = rb_check_array_type(argv[i]); + if (NIL_P(ary)) { allary = Qfalse; break; } + argv[i] = ary; } if (!allary) { CONST_ID(conv, "to_enum"); @@ -1576,7 +1580,7 @@ enum_zip(int argc, VALUE *argv, VALUE obj) result = rb_ary_new(); } /* use NODE_DOT2 as memo(v, v, -) */ - memo = rb_node_newnode(NODE_DOT2, result, rb_ary_new4(argc, argv), 0); + memo = rb_node_newnode(NODE_DOT2, result, args, 0); rb_block_call(obj, id_each, 0, 0, allary ? zip_ary : zip_i, (VALUE)memo); return result; diff --git a/test/ruby/test_array.rb b/test/ruby/test_array.rb index 9edd31dfa7..1cd018be85 100644 --- a/test/ruby/test_array.rb +++ b/test/ruby/test_array.rb @@ -1502,6 +1502,14 @@ class TestArray < Test::Unit::TestCase a = [] [1, 2, 3].zip([:a, :b], ["a", "b", "c", "d"]) {|x| a << x } assert_equal([[1, :a, "a"], [2, :b, "b"], [3, nil, "c"]], a) + + ary = Object.new + def ary.to_a; [1, 2]; end + assert_raise(NoMethodError){ %w(a b).zip(ary) } + def ary.each; [3, 4].each{|e|yield e}; end + assert_equal([['a', 3], ['b', 4]], %w(a b).zip(ary)) + def ary.to_ary; [5, 6]; end + assert_equal([['a', 5], ['b', 6]], %w(a b).zip(ary)) end def test_transpose diff --git a/test/ruby/test_enum.rb b/test/ruby/test_enum.rb index ed82d7787f..a8a88640bd 100644 --- a/test/ruby/test_enum.rb +++ b/test/ruby/test_enum.rb @@ -211,6 +211,14 @@ class TestEnumerable < Test::Unit::TestCase a = [] @obj.zip([:a, :b, :c]) {|x,y| a << [x, y] } assert_equal([[1,:a],[2,:b],[3,:c],[1,nil],[2,nil]], a) + + ary = Object.new + def ary.to_a; [1, 2]; end + assert_raise(NoMethodError){ %w(a b).zip(ary) } + def ary.each; [3, 4].each{|e|yield e}; end + assert_equal([[1, 3], [2, 4], [3, nil], [1, nil], [2, nil]], @obj.zip(ary)) + def ary.to_ary; [5, 6]; end + assert_equal([[1, 5], [2, 6], [3, nil], [1, nil], [2, nil]], @obj.zip(ary)) end def test_take |