1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

* parse.y (f_block_arg), eval.c (rb_yield_0): deal with dynamic

variable lambda arguments.  [ruby-core:05540]


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@8955 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
nobu 2005-08-08 23:24:29 +00:00
parent 81579af43e
commit 09d57b8e0c
4 changed files with 66 additions and 8 deletions

View file

@ -1,3 +1,8 @@
Tue Aug 9 08:24:05 2005 Mauricio Fernandez <mfp@acm.org>
* parse.y (f_block_arg), eval.c (rb_yield_0): deal with dynamic
variable lambda arguments. [ruby-core:05540]
Mon Aug 8 22:13:48 2005 Nobuyoshi Nakada <nobu@ruby-lang.org>
* eval.c (assign): deal with new block argument.

14
eval.c
View file

@ -4893,6 +4893,13 @@ rb_yield_0(val, self, klass, flags, avalue)
}
formal_assign(self, var, RARRAY(val)->len, RARRAY(val)->ptr, 0);
}
else if (nd_type(var) == NODE_BLOCK) {
if (var->nd_next) {
bvar = var->nd_next->nd_head;
}
var = var->nd_head;
goto block_var;
}
else {
int len = 0;
if (avalue) {
@ -5269,13 +5276,6 @@ assign(self, lhs, val, pcall)
}
break;
case NODE_BLOCK:
lhs = lhs->nd_head;
if (nd_type(lhs) == NODE_ARGS) {
formal_assign(self, lhs, 1, &val, 0);
break;
}
default:
rb_bug("bug in variable assignment");
break;

View file

@ -4192,7 +4192,7 @@ f_block_arg : blkarg_mark tIDENTIFIER
yyerror("block argument must be local variable");
else if (!dyna_in_block() && local_id($2))
yyerror("duplicated block argument name");
$$ = NEW_BLOCK_ARG($2);
$$ = dyna_in_block() ? assignable($2, 0) : NEW_BLOCK_ARG($2);
/*%
$$ = $2;
%*/

53
test/ruby/test_lambda.rb Normal file
View file

@ -0,0 +1,53 @@
require 'test/unit'
class TestLambdaParameters < Test::Unit::TestCase
def test_call_simple
assert_equal(1, ->(a){ a }.call(1))
assert_equal([1,2], ->(a,b){ [a,b] }.call(1,2))
assert_raises(ArgumentError) { ->(a){ }.call(1,2) }
assert_raises(ArgumentError) { ->(a){ }.call() }
assert_raises(ArgumentError) { ->(){ }.call(1) }
assert_raises(ArgumentError) { ->(a,b){ }.call(1,2,3) }
end
def test_call_rest_args
assert_equal([1,2], ->(*a){ a }.call(1,2))
assert_equal([1,2,[]], ->(a,b,*c){ [a,b,c] }.call(1,2))
assert_raises(ArgumentError){ ->(a,*b){ }.call() }
end
def test_call_opt_args
assert_equal([1,2,3,4], ->(a,b,c=3,d=4){ [a,b,c,d] }.call(1,2))
assert_equal([1,2,3,4], ->(a,b,c=0,d=4){ [a,b,c,d] }.call(1,2,3))
assert_raises(ArgumentError){ ->(a,b=1){ }.call() }
assert_raises(ArgumentError){ ->(a,b=1){ }.call(1,2,3) }
end
def test_call_rest_and_opt
assert_equal([1,2,3,[]], ->(a,b=2,c=3,*d){ [a,b,c,d] }.call(1))
assert_equal([1,2,3,[]], ->(a,b=0,c=3,*d){ [a,b,c,d] }.call(1,2))
assert_equal([1,2,3,[4,5,6]], ->(a,b=0,c=0,*d){ [a,b,c,d] }.call(1,2,3,4,5,6))
assert_raises(ArgumentError){ ->(a,b=1,*c){ }.call() }
end
def test_call_with_block
f = ->(a,b,c=3,*d,&e){ [a,b,c,d,e.call(d + [a,b,c])] }
assert_equal([1,2,3,[],6], f.call(1,2){|z| z.inject{|s,x| s+x} } )
assert_equal(nil, ->(&b){ b }.call)
foo { puts "bogus block " }
assert_equal(1, ->(&b){ b.call }.call { 1 })
b = nil
assert_equal(1, ->(&b){ b.call }.call { 1 })
assert_not_nil(b)
end
def foo
assert_equal(nil, ->(&b){ b }.call)
end
def test_lambda_as_iterator
a = 0
2.times ->(_){ a += 1 }
assert_equal(a, 2)
end
end