From 2ac037f54bb2588b7a86e61c9c6e9504884edbec Mon Sep 17 00:00:00 2001 From: Bram Moolenaar Date: Mon, 9 May 2022 14:12:14 +0100 Subject: patch 8.2.4927: return type of remove() incorrect when using three arguments Problem: Return type of remove() incorrect when using three arguments. Solution: Use first argument type when there are three arguments. (closes #10387) --- src/evalfunc.c | 7 ++++++- src/testdir/test_vim9_builtin.vim | 11 +++++------ src/version.c | 2 ++ 3 files changed, 13 insertions(+), 7 deletions(-) diff --git a/src/evalfunc.c b/src/evalfunc.c index efade6cab..f7a04fdc9 100644 --- a/src/evalfunc.c +++ b/src/evalfunc.c @@ -1420,13 +1420,18 @@ ret_argv(int argcount, static type_T * ret_remove(int argcount, type2_T *argtypes, - type_T **decl_type UNUSED) + type_T **decl_type) { if (argcount > 0) { if (argtypes[0].type_curr->tt_type == VAR_LIST || argtypes[0].type_curr->tt_type == VAR_DICT) { + if (argcount == 3) + { + *decl_type = argtypes[0].type_decl; + return argtypes[0].type_curr; + } if (argtypes[0].type_curr->tt_type == argtypes[0].type_decl->tt_type) *decl_type = argtypes[0].type_decl->tt_member; diff --git a/src/testdir/test_vim9_builtin.vim b/src/testdir/test_vim9_builtin.vim index c9334bc0f..2c4e9ae99 100644 --- a/src/testdir/test_vim9_builtin.vim +++ b/src/testdir/test_vim9_builtin.vim @@ -3208,12 +3208,11 @@ def Test_remove() enddef def Test_remove_return_type() - var l = remove({one: [1, 2], two: [3, 4]}, 'one') - var res = 0 - for n in l - res += n - endfor - res->assert_equal(3) + var l: list = remove({one: [1, 2], two: [3, 4]}, 'one') + l->assert_equal([1, 2]) + + var ll: list = remove(range(3), 0, 1) + ll->assert_equal([0, 1]) enddef def Test_rename() diff --git a/src/version.c b/src/version.c index c23757fe2..54eb6521b 100644 --- a/src/version.c +++ b/src/version.c @@ -746,6 +746,8 @@ static char *(features[]) = static int included_patches[] = { /* Add new patch number below this line */ +/**/ + 4927, /**/ 4926, /**/ -- cgit v1.2.1