1" Tests for ->method() 2 3func Test_list_method() 4 let l = [1, 2, 3] 5 call assert_equal([1, 2, 3, 4], [1, 2, 3]->add(4)) 6 eval l->assert_equal(l) 7 eval l->assert_equal(l, 'wrong') 8 eval l->assert_notequal([3, 2, 1]) 9 eval l->assert_notequal([3, 2, 1], 'wrong') 10 call assert_equal(l, l->copy()) 11 call assert_equal(1, l->count(2)) 12 call assert_false(l->empty()) 13 call assert_true([]->empty()) 14 call assert_equal(579, ['123', '+', '456']->join()->eval()) 15 call assert_equal([1, 2, 3, 4, 5], [1, 2, 3]->extend([4, 5])) 16 call assert_equal([1, 3], [1, 2, 3]->filter('v:val != 2')) 17 call assert_equal(2, l->get(1)) 18 call assert_equal(1, l->index(2)) 19 call assert_equal([0, 1, 2, 3], [1, 2, 3]->insert(0)) 20 call assert_fails('eval l->items()', 'E715:') 21 call assert_equal('1 2 3', l->join()) 22 call assert_fails('eval l->keys()', 'E715:') 23 call assert_equal(3, l->len()) 24 call assert_equal([2, 3, 4], [1, 2, 3]->map('v:val + 1')) 25 call assert_equal(3, l->max()) 26 call assert_equal(1, l->min()) 27 call assert_equal(2, [1, 2, 3]->remove(1)) 28 call assert_equal([1, 2, 3, 1, 2, 3], l->repeat(2)) 29 call assert_equal([3, 2, 1], [1, 2, 3]->reverse()) 30 call assert_equal([1, 2, 3, 4], [4, 2, 3, 1]->sort()) 31 call assert_equal('[1, 2, 3]', l->string()) 32 call assert_equal(v:t_list, l->type()) 33 call assert_equal([1, 2, 3], [1, 1, 2, 3, 3]->uniq()) 34 call assert_fails('eval l->values()', 'E715:') 35endfunc 36 37func Test_dict_method() 38 let d = #{one: 1, two: 2, three: 3} 39 40 call assert_equal(d, d->copy()) 41 call assert_equal(1, d->count(2)) 42 call assert_false(d->empty()) 43 call assert_true({}->empty()) 44 call assert_equal(#{one: 1, two: 2, three: 3, four: 4}, d->extend(#{four: 4})) 45 call assert_equal(#{one: 1, two: 2, three: 3}, d->filter('v:val != 4')) 46 call assert_equal(2, d->get('two')) 47 call assert_fails("let x = d->index(2)", 'E897:') 48 call assert_fails("let x = d->insert(0)", 'E899:') 49 call assert_true(d->has_key('two')) 50 call assert_equal([['one', 1], ['two', 2], ['three', 3]], d->items()) 51 call assert_fails("let x = d->join()", 'E714:') 52 call assert_equal(['one', 'two', 'three'], d->keys()) 53 call assert_equal(3, d->len()) 54 call assert_equal(#{one: 2, two: 3, three: 4}, d->map('v:val + 1')) 55 call assert_equal(#{one: 1, two: 2, three: 3}, d->map('v:val - 1')) 56 call assert_equal(3, d->max()) 57 call assert_equal(1, d->min()) 58 call assert_equal(2, d->remove("two")) 59 let d.two = 2 60 call assert_fails('let x = d->repeat(2)', 'E731:') 61 call assert_fails('let x = d->reverse()', 'E899:') 62 call assert_fails('let x = d->sort()', 'E686:') 63 call assert_equal("{'one': 1, 'two': 2, 'three': 3}", d->string()) 64 call assert_equal(v:t_dict, d->type()) 65 call assert_fails('let x = d->uniq()', 'E686:') 66 call assert_equal([1, 2, 3], d->values()) 67endfunc 68 69func Test_string_method() 70 call assert_equal(['1', '2', '3'], '1 2 3'->split()) 71 call assert_equal([1, 2, 3], '1 2 3'->split()->map({i, v -> str2nr(v)})) 72 call assert_equal([65, 66, 67], 'ABC'->str2list()) 73 call assert_equal(3, 'ABC'->strlen()) 74 call assert_equal('a^Mb^[c', "a\rb\ec"->strtrans()) 75 call assert_equal(4, "aあb"->strwidth()) 76 call assert_equal('axc', 'abc'->substitute('b', 'x', '')) 77endfunc 78 79func Test_method_append() 80 new 81 eval ['one', 'two', 'three']->append(1) 82 call assert_equal(['', 'one', 'two', 'three'], getline(1, '$')) 83 84 %del 85 let bnr = bufnr('') 86 wincmd w 87 eval ['one', 'two', 'three']->appendbufline(bnr, 1) 88 call assert_equal(['', 'one', 'two', 'three'], getbufline(bnr, 1, '$')) 89 90 exe 'bwipe! ' .. bnr 91endfunc 92 93func Test_method_funcref() 94 func Concat(one, two, three) 95 return a:one .. a:two .. a:three 96 endfunc 97 let FuncRef = function('Concat') 98 eval 'foo'->FuncRef('bar', 'tail')->assert_equal('foobartail') 99 100 let Partial = function('Concat', ['two']) 101 eval 'one'->Partial('three')->assert_equal('onetwothree') 102 103 delfunc Concat 104endfunc 105