1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181
|
# frozen_string_literal: false
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_raise(ArgumentError) { lambda{|a|}.call(1,2) }
assert_raise(ArgumentError) { lambda{|a|}.call() }
assert_raise(ArgumentError) { lambda{}.call(1) }
assert_raise(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_raise(ArgumentError) { ->(a){ }.call(1,2) }
assert_raise(ArgumentError) { ->(a){ }.call() }
assert_raise(ArgumentError) { ->(){ }.call(1) }
assert_raise(ArgumentError) { ->(a,b){ }.call(1,2,3) }
end
def test_lambda_as_iterator
a = 0
2.times(&->(_){ a += 1 })
assert_equal(2, a)
assert_raise(ArgumentError) {1.times(&->(){ a += 1 })}
bug9605 = '[ruby-core:61468] [Bug #9605]'
assert_nothing_raised(ArgumentError, bug9605) {1.times(&->(n){ a += 1 })}
assert_equal(3, a, bug9605)
assert_nothing_raised(ArgumentError, bug9605) {a = [[1, 2]].map(&->(x, y) {x+y})}
assert_equal([3], a, bug9605)
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_raise(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_raise(ArgumentError){ ->(a,b=1){ }.call() }
assert_raise(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_raise(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 test_call_block_from_lambda
bug9605 = '[ruby-core:61470] [Bug #9605]'
plus = ->(x,y) {x+y}
assert_raise(ArgumentError, bug9605) {proc(&plus).call [1,2]}
end
def test_instance_exec
bug12568 = '[ruby-core:76300] [Bug #12568]'
assert_nothing_raised(ArgumentError, bug12568) do
instance_exec([1,2,3], &->(a=[]){ a })
end
end
def yield_1(arg)
yield arg
end
tap do |;bug9605, expected, result|
bug9605 = '[ruby-core:65887] [Bug #9605] arity check should be relaxed'
expected = [1,2,3]
[
["array", expected],
["to_ary", Struct.new(:to_ary).new(expected)],
].product \
[
["proc", proc {|a, b, c| [a, b, c]}],
["lambda", lambda {|a, b, c| [a, b, c]}],
] do
|(vtype, val), (btype, block)|
define_method("test_yield_relaxed(#{vtype},&#{btype})") do
result = assert_nothing_raised(ArgumentError, bug9605) {
break yield_1(val, &block)
}
assert_equal(expected, result, bug9605)
end
end
end
def foo
assert_equal(nil, ->(&b){ b }.call)
end
def test_in_basic_object
bug5966 = '[ruby-core:42349]'
called = false
BasicObject.new.instance_eval {->() {called = true}.()}
assert_equal(true, called, bug5966)
end
def test_location_on_error
bug6151 = '[ruby-core:43314]'
called = 0
line, f = __LINE__, lambda do
called += 1
true
end
e = assert_raise(ArgumentError) do
f.call(42)
end
assert_send([e.backtrace.first, :start_with?, "#{__FILE__}:#{line}:"], bug6151)
assert_equal(0, called)
e = assert_raise(ArgumentError) do
42.times(&f)
end
assert_send([e.backtrace.first, :start_with?, "#{__FILE__}:#{line}:"], bug6151)
assert_equal(0, called)
end
def return_in_current(val)
1.tap(&->(*) {return 0})
val
end
def yield_block
yield
end
def return_in_callee(val)
yield_block(&->(*) {return 0})
val
end
def test_return
feature8693 = '[ruby-core:56193] [Feature #8693]'
assert_equal(42, return_in_current(42), feature8693)
assert_equal(42, return_in_callee(42), feature8693)
end
def test_do_lambda_source_location
exp_lineno = __LINE__ + 3
lmd = ->(x,
y,
z) do
#
end
file, lineno = lmd.source_location
assert_match(/^#{ Regexp.quote(__FILE__) }$/, file)
assert_equal(exp_lineno, lineno, "must be at the beginning of the block")
end
def test_brace_lambda_source_location
exp_lineno = __LINE__ + 3
lmd = ->(x,
y,
z) {
#
}
file, lineno = lmd.source_location
assert_match(/^#{ Regexp.quote(__FILE__) }$/, file)
assert_equal(exp_lineno, lineno, "must be at the beginning of the block")
end
end
|