mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
78029f00d5
passing parameter. * eval.c, eval_jump.h: simplify rb_yield*. * proc.c (proc_mark): fix to mark proc->block.proc. * proc.c (Init_Proc): add Proc#lambda? * test/ruby/test_lambda.rb: add some tests. * vm.c (invoke_block): fix to check lambda block or not. * vm.c (th_yield_setup_args): fix to check arguments size when lambda block. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@12441 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
68 lines
2.1 KiB
Ruby
68 lines
2.1 KiB
Ruby
require 'test/unit'
|
|
|
|
class TestLambdaParameters < Test::Unit::TestCase
|
|
|
|
def test_exact_parameter
|
|
assert_raise(ArgumentError){(1..3).each(&lambda{})}
|
|
end
|
|
|
|
def test_call_simple
|
|
assert_equal(1, lambda{|a| a}.call(1))
|
|
assert_equal([1,2], lambda{|a, b| [a,b]}.call(1,2))
|
|
assert_raises(ArgumentError) { lambda{|a|}.call(1,2) }
|
|
assert_raises(ArgumentError) { lambda{|a|}.call() }
|
|
assert_raises(ArgumentError) { lambda{}.call(1) }
|
|
assert_raises(ArgumentError) { lambda{|a, b|}.call(1,2,3) }
|
|
|
|
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
|
|
|
|
end
|
|
|
|
__END__
|
|
def test_lambda_as_iterator
|
|
a = 0
|
|
2.times(&->(_){ a += 1 })
|
|
assert_equal(a, 2)
|
|
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_nil(b)
|
|
end
|
|
|
|
def foo
|
|
assert_equal(nil, ->(&b){ b }.call)
|
|
end
|
|
end
|