ruby-changes:10134
From: nobu <ko1@a...>
Date: Tue, 20 Jan 2009 06:48:10 +0900 (JST)
Subject: [ruby-changes:10134] Ruby:r21678 (trunk): * array.c (take_items), enum.c (enum_zip): tries to convert to
nobu 2009-01-20 06:47:48 +0900 (Tue, 20 Jan 2009) New Revision: 21678 http://svn.ruby-lang.org/cgi-bin/viewvc.cgi?view=rev&revision=21678 Log: * array.c (take_items), enum.c (enum_zip): tries to convert to array first. [ruby-core:21442] Modified files: trunk/ChangeLog trunk/array.c trunk/enum.c trunk/test/ruby/test_array.rb trunk/test/ruby/test_enum.rb Index: array.c =================================================================== --- array.c (revision 21677) +++ array.c (revision 21678) @@ -2209,9 +2209,11 @@ static VALUE take_items(VALUE obj, long n) { - VALUE result = rb_ary_new2(n); + VALUE result = to_ary(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; Index: ChangeLog =================================================================== --- ChangeLog (revision 21677) +++ ChangeLog (revision 21678) @@ -1,3 +1,8 @@ +Tue Jan 20 06:48:56 2009 Nobuyoshi Nakada <nobu@r...> + + * array.c (take_items), enum.c (enum_zip): tries to convert to + array first. [ruby-core:21442] + Tue Jan 20 03:50:37 2009 NARUSE, Yui <naruse@r...> * ext/nkf/nkf-utf8/nkf.c: Update nkf to 2.0.9. Index: enum.c =================================================================== --- enum.c (revision 21677) +++ enum.c (revision 21678) @@ -1558,13 +1558,17 @@ 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 @@ 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; Index: test/ruby/test_array.rb =================================================================== --- test/ruby/test_array.rb (revision 21677) +++ test/ruby/test_array.rb (revision 21678) @@ -1502,6 +1502,11 @@ 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 + def ary.to_ary; [3, 4]; end + assert_equal([[5, 3], [6, 4]], [5, 6].zip(ary)) end def test_transpose Index: test/ruby/test_enum.rb =================================================================== --- test/ruby/test_enum.rb (revision 21677) +++ test/ruby/test_enum.rb (revision 21678) @@ -211,6 +211,11 @@ 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 + def ary.to_ary; [3, 4]; end + assert_equal([[1, 3], [2, 4], [3, nil], [1, nil], [2, nil]], @obj.zip(ary)) end def test_take -- ML: ruby-changes@q... Info: http://www.atdot.net/~ko1/quickml/