mirror of
https://github.com/vim/vim.git
synced 2025-10-27 09:24:23 -04:00
Problem: tests: too many imports in the test suite Solution: Clean up the imported scripts Most tests make use of check.vim, so let's just source it once in runtest.vim instead of having each test manually source it. runtest.vim already sources shared.vim, which again sources view_util.vim, so we don't need to source those two common dependencies in all the other tests And then check.vim sources term_util.vim already, so we can in addition drop sourcing it explicitly in each single test script. Note: test_expand_func.vim had to be updated to account for the changed number of sourced files. And finally check.vim uses line-continuation so let's also explicitly enable line continuation via the 'cpo' option value. related: #17677 Signed-off-by: Christian Brabandt <cb@256bit.org>
384 lines
8.2 KiB
VimL
384 lines
8.2 KiB
VimL
" Test for lambda and closure
|
|
|
|
import './vim9.vim' as v9
|
|
|
|
func Test_lambda_feature()
|
|
call assert_equal(1, has('lambda'))
|
|
endfunc
|
|
|
|
func Test_lambda_with_filter()
|
|
let s:x = 2
|
|
call assert_equal([2, 3], filter([1, 2, 3], {i, v -> v >= s:x}))
|
|
endfunc
|
|
|
|
func Test_lambda_with_map()
|
|
let s:x = 1
|
|
call assert_equal([2, 3, 4], map([1, 2, 3], {i, v -> v + s:x}))
|
|
endfunc
|
|
|
|
func Test_lambda_with_sort()
|
|
call assert_equal([1, 2, 3, 4, 7], sort([3,7,2,1,4], {a, b -> a - b}))
|
|
endfunc
|
|
|
|
func Test_lambda_with_timer()
|
|
CheckFeature timers
|
|
|
|
let s:n = 0
|
|
let s:timer_id = 0
|
|
func! s:Foo()
|
|
let s:timer_id = timer_start(10, {-> execute("let s:n += 1 | echo s:n", "")}, {"repeat": -1})
|
|
endfunc
|
|
|
|
call s:Foo()
|
|
" check timer works
|
|
for i in range(0, 10)
|
|
if s:n > 0
|
|
break
|
|
endif
|
|
sleep 10m
|
|
endfor
|
|
|
|
" do not collect lambda
|
|
call test_garbagecollect_now()
|
|
|
|
" check timer still works
|
|
let m = s:n
|
|
for i in range(0, 10)
|
|
if s:n > m
|
|
break
|
|
endif
|
|
sleep 10m
|
|
endfor
|
|
|
|
call timer_stop(s:timer_id)
|
|
call assert_true(s:n > m)
|
|
endfunc
|
|
|
|
func Test_lambda_vim9cmd_linebreak()
|
|
CheckFeature timers
|
|
|
|
let g:test_is_flaky = 1
|
|
let lines =<< trim END
|
|
vim9cmd call timer_start(10, (x) => {
|
|
# comment
|
|
g:result = 'done'
|
|
})
|
|
END
|
|
call v9.CheckScriptSuccess(lines)
|
|
" sleep longer on a retry
|
|
exe 'sleep ' .. [20, 100, 500, 500, 500][g:run_nr] .. 'm'
|
|
call assert_equal('done', g:result)
|
|
unlet g:result
|
|
|
|
let lines =<< trim END
|
|
g:result = [0]->map((_, v) =>
|
|
1 # inline comment
|
|
+
|
|
2
|
|
)
|
|
assert_equal([3], g:result)
|
|
END
|
|
call v9.CheckDefAndScriptSuccess(lines)
|
|
endfunc
|
|
|
|
def Test_lamba_compiled_linebreak()
|
|
var lines =<< trim END
|
|
vim9script
|
|
|
|
def Echo(what: any)
|
|
assert_equal('hello world', what)
|
|
enddef
|
|
def That()
|
|
printf("hello ")
|
|
->((x) => x .. "world")()
|
|
->Echo()
|
|
enddef
|
|
That()
|
|
END
|
|
v9.CheckScriptSuccess(lines)
|
|
enddef
|
|
|
|
func Test_lambda_with_partial()
|
|
let l:Cb = function({... -> ['zero', a:1, a:2, a:3]}, ['one', 'two'])
|
|
call assert_equal(['zero', 'one', 'two', 'three'], l:Cb('three'))
|
|
endfunc
|
|
|
|
function Test_lambda_fails()
|
|
call assert_equal(3, {a, b -> a + b}(1, 2))
|
|
call assert_fails('echo {a, a -> a + a}(1, 2)', 'E853:')
|
|
call assert_fails('echo {a, b -> a + b)}(1, 2)', 'E451:')
|
|
echo assert_fails('echo 10->{a -> a + 2}', 'E107:')
|
|
|
|
call assert_fails('eval 0->(', "E110: Missing ')'")
|
|
call assert_fails('eval 0->(3)()', "E1275:")
|
|
call assert_fails('eval 0->([3])()', "E1275:")
|
|
call assert_fails('eval 0->({"a": 3})()', "E1275:")
|
|
call assert_fails('eval 0->(xxx)()', "E121:")
|
|
endfunc
|
|
|
|
func Test_not_lamda()
|
|
let x = {'>' : 'foo'}
|
|
call assert_equal('foo', x['>'])
|
|
endfunc
|
|
|
|
func Test_lambda_capture_by_reference()
|
|
let v = 1
|
|
let l:F = {x -> x + v}
|
|
let v = 2
|
|
call assert_equal(12, l:F(10))
|
|
endfunc
|
|
|
|
func Test_lambda_side_effect()
|
|
func! s:update_and_return(arr)
|
|
let a:arr[1] = 5
|
|
return a:arr
|
|
endfunc
|
|
|
|
func! s:foo(arr)
|
|
return {-> s:update_and_return(a:arr)}
|
|
endfunc
|
|
|
|
let arr = [3,2,1]
|
|
call assert_equal([3, 5, 1], s:foo(arr)())
|
|
endfunc
|
|
|
|
func Test_lambda_refer_local_variable_from_other_scope()
|
|
func! s:foo(X)
|
|
return a:X() " refer l:x in s:bar()
|
|
endfunc
|
|
|
|
func! s:bar()
|
|
let x = 123
|
|
return s:foo({-> x})
|
|
endfunc
|
|
|
|
call assert_equal(123, s:bar())
|
|
endfunc
|
|
|
|
func Test_lambda_do_not_share_local_variable()
|
|
func! s:define_funcs()
|
|
let l:One = {-> split(execute("let a = 'abc' | echo a"))[0]}
|
|
let l:Two = {-> exists("a") ? a : "no"}
|
|
return [l:One, l:Two]
|
|
endfunc
|
|
|
|
let l:F = s:define_funcs()
|
|
|
|
call assert_equal('no', l:F[1]())
|
|
call assert_equal('abc', l:F[0]())
|
|
call assert_equal('no', l:F[1]())
|
|
endfunc
|
|
|
|
func Test_lambda_closure_counter()
|
|
func! s:foo()
|
|
let x = 0
|
|
return {-> [execute("let x += 1"), x][-1]}
|
|
endfunc
|
|
|
|
let l:F = s:foo()
|
|
call test_garbagecollect_now()
|
|
call assert_equal(1, l:F())
|
|
call assert_equal(2, l:F())
|
|
call assert_equal(3, l:F())
|
|
call assert_equal(4, l:F())
|
|
endfunc
|
|
|
|
func Test_lambda_with_a_var()
|
|
func! s:foo()
|
|
let x = 2
|
|
return {... -> a:000 + [x]}
|
|
endfunc
|
|
func! s:bar()
|
|
return s:foo()(1)
|
|
endfunc
|
|
|
|
call assert_equal([1, 2], s:bar())
|
|
endfunc
|
|
|
|
func Test_lambda_call_lambda_from_lambda()
|
|
func! s:foo(x)
|
|
let l:F1 = {-> {-> a:x}}
|
|
return {-> l:F1()}
|
|
endfunc
|
|
|
|
let l:F = s:foo(1)
|
|
call assert_equal(1, l:F()())
|
|
endfunc
|
|
|
|
func Test_lambda_delfunc()
|
|
func! s:gen()
|
|
let pl = l:
|
|
let l:Foo = {-> get(pl, "Foo", get(pl, "Bar", {-> 0}))}
|
|
let l:Bar = l:Foo
|
|
delfunction l:Foo
|
|
return l:Bar
|
|
endfunc
|
|
|
|
let l:F = s:gen()
|
|
call assert_fails(':call l:F()', 'E933:')
|
|
endfunc
|
|
|
|
func Test_lambda_scope()
|
|
func! s:NewCounter()
|
|
let c = 0
|
|
return {-> [execute('let c += 1'), c][-1]}
|
|
endfunc
|
|
|
|
func! s:NewCounter2()
|
|
return {-> [execute('let c += 100'), c][-1]}
|
|
endfunc
|
|
|
|
let l:C = s:NewCounter()
|
|
let l:D = s:NewCounter2()
|
|
|
|
call assert_equal(1, l:C())
|
|
call assert_fails(':call l:D()', 'E121:')
|
|
call assert_equal(2, l:C())
|
|
endfunc
|
|
|
|
func Test_lambda_share_scope()
|
|
func! s:New()
|
|
let c = 0
|
|
let l:Inc0 = {-> [execute('let c += 1'), c][-1]}
|
|
let l:Dec0 = {-> [execute('let c -= 1'), c][-1]}
|
|
return [l:Inc0, l:Dec0]
|
|
endfunc
|
|
|
|
let [l:Inc, l:Dec] = s:New()
|
|
|
|
call assert_equal(1, l:Inc())
|
|
call assert_equal(2, l:Inc())
|
|
call assert_equal(1, l:Dec())
|
|
endfunc
|
|
|
|
func Test_lambda_circular_reference()
|
|
func! s:Foo()
|
|
let d = {}
|
|
let d.f = {-> d}
|
|
return d.f
|
|
endfunc
|
|
|
|
call s:Foo()
|
|
call test_garbagecollect_now()
|
|
let i = 0 | while i < 10000 | call s:Foo() | let i+= 1 | endwhile
|
|
call test_garbagecollect_now()
|
|
endfunc
|
|
|
|
func Test_lambda_combination()
|
|
call assert_equal(2, {x -> {x -> x}}(1)(2))
|
|
call assert_equal(10, {y -> {x -> x(y)(10)}({y -> y})}({z -> z}))
|
|
call assert_equal(5.0, {x -> {y -> x / y}}(10)(2.0))
|
|
call assert_equal(6, {x -> {y -> {z -> x + y + z}}}(1)(2)(3))
|
|
|
|
call assert_equal(6, {x -> {f -> f(x)}}(3)({x -> x * 2}))
|
|
call assert_equal(6, {f -> {x -> f(x)}}({x -> x * 2})(3))
|
|
|
|
" Z combinator
|
|
let Z = {f -> {x -> f({y -> x(x)(y)})}({x -> f({y -> x(x)(y)})})}
|
|
let Fact = {f -> {x -> x == 0 ? 1 : x * f(x - 1)}}
|
|
call assert_equal(120, Z(Fact)(5))
|
|
endfunc
|
|
|
|
func Test_closure_counter()
|
|
func! s:foo()
|
|
let x = 0
|
|
func! s:bar() closure
|
|
let x += 1
|
|
return x
|
|
endfunc
|
|
return function('s:bar')
|
|
endfunc
|
|
|
|
let l:F = s:foo()
|
|
call test_garbagecollect_now()
|
|
call assert_equal(1, l:F())
|
|
call assert_equal(2, l:F())
|
|
call assert_equal(3, l:F())
|
|
call assert_equal(4, l:F())
|
|
|
|
call assert_match("^\n function <SNR>\\d\\+_bar() closure"
|
|
\ .. "\n1 let x += 1"
|
|
\ .. "\n2 return x"
|
|
\ .. "\n endfunction$", execute('func s:bar'))
|
|
endfunc
|
|
|
|
func Test_closure_unlet()
|
|
func! s:foo()
|
|
let x = 1
|
|
func! s:bar() closure
|
|
unlet x
|
|
endfunc
|
|
call s:bar()
|
|
return l:
|
|
endfunc
|
|
|
|
call assert_false(has_key(s:foo(), 'x'))
|
|
call test_garbagecollect_now()
|
|
endfunc
|
|
|
|
func LambdaFoo()
|
|
let x = 0
|
|
func! LambdaBar() closure
|
|
let x += 1
|
|
return x
|
|
endfunc
|
|
return function('LambdaBar')
|
|
endfunc
|
|
|
|
func Test_closure_refcount()
|
|
let g:Count = LambdaFoo()
|
|
call test_garbagecollect_now()
|
|
call assert_equal(1, g:Count())
|
|
let g:Count2 = LambdaFoo()
|
|
call test_garbagecollect_now()
|
|
call assert_equal(1, g:Count2())
|
|
call assert_equal(2, g:Count())
|
|
call assert_equal(3, g:Count2())
|
|
|
|
delfunc LambdaFoo
|
|
delfunc LambdaBar
|
|
endfunc
|
|
|
|
func Test_named_function_closure()
|
|
func! Afoo()
|
|
let x = 14
|
|
func! s:Abar() closure
|
|
return x
|
|
endfunc
|
|
call assert_equal(14, s:Abar())
|
|
endfunc
|
|
call Afoo()
|
|
call assert_equal(14, s:Abar())
|
|
call test_garbagecollect_now()
|
|
call assert_equal(14, s:Abar())
|
|
endfunc
|
|
|
|
func Test_lambda_with_index()
|
|
let List = {x -> [x]}
|
|
let Extract = {-> function(List, ['foobar'])()[0]}
|
|
call assert_equal('foobar', Extract())
|
|
endfunc
|
|
|
|
func Test_lambda_error()
|
|
" This was causing a crash
|
|
call assert_fails('ec{@{->{d->()()', 'E451:')
|
|
endfunc
|
|
|
|
func Test_closure_error()
|
|
let l =<< trim END
|
|
func F1() closure
|
|
return 1
|
|
endfunc
|
|
END
|
|
call writefile(l, 'Xscript', 'D')
|
|
let caught_932 = 0
|
|
try
|
|
source Xscript
|
|
catch /E932:/
|
|
let caught_932 = 1
|
|
endtry
|
|
call assert_equal(1, caught_932)
|
|
endfunc
|
|
|
|
" vim: shiftwidth=2 sts=2 expandtab
|