1" Tests for the List and Dict types
2
3func TearDown()
4  " Run garbage collection after every test
5  call test_garbagecollect_now()
6endfunc
7
8" Tests for List type
9
10" List creation
11func Test_list_create()
12  " Creating List directly with different types
13  let l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
14  call assert_equal("[1, 'as''d', [1, 2, function('strlen')], {'a': 1}]", string(l))
15  call assert_equal({'a' : 1}, l[-1])
16  call assert_equal(1, l[-4])
17  let x = 10
18  try
19    let x = l[-5]
20  catch
21    call assert_match('E684:', v:exception)
22  endtry
23  call assert_equal(10, x)
24endfunc
25
26" This was allowed in legacy Vim script
27let s:list_with_spaces = [1 , 2 , 3]
28
29" List slices
30func Test_list_slice()
31  let l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
32  call assert_equal([1, 'as''d', [1, 2, function('strlen')], {'a': 1}], l[:])
33  call assert_equal(['as''d', [1, 2, function('strlen')], {'a': 1}], l[1:])
34  call assert_equal([1, 'as''d', [1, 2, function('strlen')]], l[:-2])
35  call assert_equal([1, 'as''d', [1, 2, function('strlen')], {'a': 1}], l[0:8])
36  call assert_equal([], l[8:-1])
37  call assert_equal([], l[0:-10])
38  " perform an operation on a list slice
39  let l = [1, 2, 3]
40  let l[:1] += [1, 2]
41  let l[2:] -= [1]
42  call assert_equal([2, 4, 2], l)
43endfunc
44
45" List identity
46func Test_list_identity()
47  let l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
48  let ll = l
49  let lx = copy(l)
50  call assert_true(l == ll)
51  call assert_false(l isnot ll)
52  call assert_true(l is ll)
53  call assert_true(l == lx)
54  call assert_false(l is lx)
55  call assert_true(l isnot lx)
56endfunc
57
58" removing items with :unlet
59func Test_list_unlet()
60  let l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
61  unlet l[2]
62  call assert_equal([1, 'as''d', {'a': 1}], l)
63  let l = range(8)
64  unlet l[:3]
65  unlet l[1:]
66  call assert_equal([4], l)
67
68  " removing items out of range: silently skip items that don't exist
69  let l = [0, 1, 2, 3]
70  call assert_fails('unlet l[2:1]', 'E684:')
71  let l = [0, 1, 2, 3]
72  unlet l[2:2]
73  call assert_equal([0, 1, 3], l)
74  let l = [0, 1, 2, 3]
75  unlet l[2:3]
76  call assert_equal([0, 1], l)
77  let l = [0, 1, 2, 3]
78  unlet l[2:4]
79  call assert_equal([0, 1], l)
80  let l = [0, 1, 2, 3]
81  unlet l[2:5]
82  call assert_equal([0, 1], l)
83  let l = [0, 1, 2, 3]
84  call assert_fails('unlet l[-1:2]', 'E684:')
85  let l = [0, 1, 2, 3]
86  unlet l[-2:2]
87  call assert_equal([0, 1, 3], l)
88  let l = [0, 1, 2, 3]
89  unlet l[-3:2]
90  call assert_equal([0, 3], l)
91  let l = [0, 1, 2, 3]
92  unlet l[-4:2]
93  call assert_equal([3], l)
94  let l = [0, 1, 2, 3]
95  unlet l[-5:2]
96  call assert_equal([3], l)
97  let l = [0, 1, 2, 3]
98  unlet l[-6:2]
99  call assert_equal([3], l)
100endfunc
101
102" assignment to a list
103func Test_list_assign()
104  let l = [0, 1, 2, 3]
105  let [va, vb] = l[2:3]
106  call assert_equal([2, 3], [va, vb])
107  call assert_fails('let [va, vb] = l', 'E687:')
108  call assert_fails('let [va, vb] = l[1:1]', 'E688:')
109endfunc
110
111" test for range assign
112func Test_list_range_assign()
113  let l = [0]
114  let l[:] = [1, 2]
115  call assert_equal([1, 2], l)
116  let l[-4:-1] = [5, 6]
117  call assert_equal([5, 6], l)
118endfunc
119
120" Test removing items in list
121func Test_list_func_remove()
122  " Test removing 1 element
123  let l = [1, 2, 3, 4]
124  call assert_equal(1, remove(l, 0))
125  call assert_equal([2, 3, 4], l)
126
127  let l = [1, 2, 3, 4]
128  call assert_equal(2, remove(l, 1))
129  call assert_equal([1, 3, 4], l)
130
131  let l = [1, 2, 3, 4]
132  call assert_equal(4, remove(l, -1))
133  call assert_equal([1, 2, 3], l)
134
135  " Test removing range of element(s)
136  let l = [1, 2, 3, 4]
137  call assert_equal([3], remove(l, 2, 2))
138  call assert_equal([1, 2, 4], l)
139
140  let l = [1, 2, 3, 4]
141  call assert_equal([2, 3], remove(l, 1, 2))
142  call assert_equal([1, 4], l)
143
144  let l = [1, 2, 3, 4]
145  call assert_equal([2, 3], remove(l, -3, -2))
146  call assert_equal([1, 4], l)
147
148  " Test invalid cases
149  let l = [1, 2, 3, 4]
150  call assert_fails("call remove(l, 5)", 'E684:')
151  call assert_fails("call remove(l, 1, 5)", 'E684:')
152  call assert_fails("call remove(l, 3, 2)", 'E16:')
153  call assert_fails("call remove(1, 0)", 'E896:')
154  call assert_fails("call remove(l, l)", 'E745:')
155endfunc
156
157" List add() function
158func Test_list_add()
159  let l = []
160  call add(l, 1)
161  call add(l, [2, 3])
162  call add(l, [])
163  call add(l, test_null_list())
164  call add(l, {'k' : 3})
165  call add(l, {})
166  call add(l, test_null_dict())
167  call assert_equal([1, [2, 3], [], [], {'k' : 3}, {}, {}], l)
168  call assert_equal(1, add(test_null_list(), 4))
169endfunc
170
171" Tests for Dictionary type
172
173func Test_dict()
174  " Creating Dictionary directly with different types
175  let d = {001: 'asd', 'b': [1, 2, function('strlen')], -1: {'a': 1},}
176  call assert_equal("{'1': 'asd', 'b': [1, 2, function('strlen')], '-1': {'a': 1}}", string(d))
177  call assert_equal('asd', d.1)
178  call assert_equal(['-1', '1', 'b'], sort(keys(d)))
179  call assert_equal(['asd', [1, 2, function('strlen')], {'a': 1}], values(d))
180  let v = []
181  for [key, val] in items(d)
182    call extend(v, [key, val])
183    unlet key val
184  endfor
185  call assert_equal(['1','asd','b',[1, 2, function('strlen')],'-1',{'a': 1}], v)
186
187  call extend(d, {3:33, 1:99})
188  call extend(d, {'b':'bbb', 'c':'ccc'}, "keep")
189  call assert_fails("call extend(d, {3:333,4:444}, 'error')", 'E737:')
190  call assert_equal({'c': 'ccc', '1': 99, 'b': [1, 2, function('strlen')], '3': 33, '-1': {'a': 1}}, d)
191  call filter(d, 'v:key =~ ''[ac391]''')
192  call assert_equal({'c': 'ccc', '1': 99, '3': 33, '-1': {'a': 1}}, d)
193
194  " duplicate key
195  call assert_fails("let d = {'k' : 10, 'k' : 20}", 'E721:')
196  " missing comma
197  call assert_fails("let d = {'k' : 10 'k' : 20}", 'E722:')
198  " missing curly brace
199  call assert_fails("let d = {'k' : 10,", 'E723:')
200  " invalid key
201  call assert_fails('let d = #{++ : 10}', 'E15:')
202  " wrong type for key
203  call assert_fails('let d={[] : 10}', 'E730:')
204  " undefined variable as value
205  call assert_fails("let d={'k' : i}", 'E121:')
206endfunc
207
208" This was allowed in legacy Vim script
209let s:dict_with_spaces = {'one' : 1 , 'two' : 2 , 'three' : 3}
210let s:dict_with_spaces_lit = #{one : 1 , two : 2 , three : 3}
211
212" Dictionary identity
213func Test_dict_identity()
214  let d = {001: 'asd', 'b': [1, 2, function('strlen')], -1: {'a': 1},}
215  let dd = d
216  let dx = copy(d)
217  call assert_true(d == dd)
218  call assert_false(d isnot dd)
219  call assert_true(d is dd)
220  call assert_true(d == dx)
221  call assert_false(d is dx)
222  call assert_true(d isnot dx)
223endfunc
224
225" removing items with :unlet
226func Test_dict_unlet()
227  let d = {'b':'bbb', '1': 99, '3': 33, '-1': {'a': 1}}
228  unlet d.b
229  unlet d[-1]
230  call assert_equal({'1': 99, '3': 33}, d)
231endfunc
232
233" manipulating a big Dictionary (hashtable.c has a border of 1000 entries)
234func Test_dict_big()
235  let d = {}
236  for i in range(1500)
237    let d[i] = 3000 - i
238  endfor
239  call assert_equal([3000, 2900, 2001, 1600, 1501], [d[0], d[100], d[999], d[1400], d[1499]])
240  let str = ''
241  try
242    let n = d[1500]
243  catch
244    let str = substitute(v:exception, '\v(.{14}).*( "\d{4}").*', '\1\2', '')
245  endtry
246  call assert_equal('Vim(let):E716: "1500"', str)
247
248  " lookup each items
249  for i in range(1500)
250    call assert_equal(3000 - i, d[i])
251  endfor
252  let i += 1
253
254  " delete even items
255  while i >= 2
256    let i -= 2
257    unlet d[i]
258  endwhile
259  call assert_equal('NONE', get(d, 1500 - 100, 'NONE'))
260  call assert_equal(2999, d[1])
261
262  " delete odd items, checking value, one intentionally wrong
263  let d[33] = 999
264  let i = 1
265  while i < 1500
266   if i != 33
267     call assert_equal(3000 - i, d[i])
268   else
269     call assert_equal(999, d[i])
270   endif
271   unlet d[i]
272   let i += 2
273  endwhile
274  call assert_equal({}, d)
275  unlet d
276endfunc
277
278" Dictionary function
279func Test_dict_func()
280  let d = {}
281  func d.func(a) dict
282    return a:a . len(self.data)
283  endfunc
284  let d.data = [1,2,3]
285  call assert_equal('len: 3', d.func('len: '))
286  let x = d.func('again: ')
287  call assert_equal('again: 3', x)
288  let Fn = d.func
289  call assert_equal('xxx3', Fn('xxx'))
290endfunc
291
292func Test_dict_assign()
293  let d = {}
294  let d.1 = 1
295  let d._ = 2
296  call assert_equal({'1': 1, '_': 2}, d)
297endfunc
298
299" Function in script-local List or Dict
300func Test_script_local_dict_func()
301  let g:dict = {}
302  function g:dict.func() dict
303    return 'g:dict.func' . self.foo[1] . self.foo[0]('asdf')
304  endfunc
305  let g:dict.foo = ['-', 2, 3]
306  call insert(g:dict.foo, function('strlen'))
307  call assert_equal('g:dict.func-4', g:dict.func())
308  unlet g:dict
309endfunc
310
311" Test removing items in a dictionary
312func Test_dict_func_remove()
313  let d = {1:'a', 2:'b', 3:'c'}
314  call assert_equal('b', remove(d, 2))
315  call assert_equal({1:'a', 3:'c'}, d)
316
317  call assert_fails("call remove(d, 1, 2)", 'E118:')
318  call assert_fails("call remove(d, 'a')", 'E716:')
319  call assert_fails("call remove(d, [])", 'E730:')
320endfunc
321
322" Nasty: remove func from Dict that's being called (works)
323func Test_dict_func_remove_in_use()
324  let d = {1:1}
325  func d.func(a)
326    return "a:" . a:a
327  endfunc
328  let expected = 'a:' . string(get(d, 'func'))
329  call assert_equal(expected, d.func(string(remove(d, 'func'))))
330endfunc
331
332func Test_dict_literal_keys()
333  call assert_equal({'one': 1, 'two2': 2, '3three': 3, '44': 4}, #{one: 1, two2: 2, 3three: 3, 44: 4},)
334
335  " why *{} cannot be used
336  let blue = 'blue'
337  call assert_equal('6', trim(execute('echo 2 *{blue: 3}.blue')))
338endfunc
339
340" Nasty: deepcopy() dict that refers to itself (fails when noref used)
341func Test_dict_deepcopy()
342  let d = {1:1, 2:2}
343  let l = [4, d, 6]
344  let d[3] = l
345  let dc = deepcopy(d)
346  call assert_fails('call deepcopy(d, 1)', 'E698:')
347  let l2 = [0, l, l, 3]
348  let l[1] = l2
349  let l3 = deepcopy(l2)
350  call assert_true(l3[1] is l3[2])
351  call assert_fails("call deepcopy([1, 2], 2)", 'E1023:')
352endfunc
353
354" Locked variables
355func Test_list_locked_var()
356  let expected = [
357	      \ [['1000-000', 'ppppppF'],
358	      \  ['0000-000', 'ppppppp'],
359	      \  ['0000-000', 'ppppppp']],
360	      \ [['1000-000', 'ppppppF'],
361	      \  ['0000-000', 'ppppppp'],
362	      \  ['0000-000', 'ppppppp']],
363	      \ [['1100-100', 'ppFppFF'],
364	      \  ['0000-000', 'ppppppp'],
365	      \  ['0000-000', 'ppppppp']],
366	      \ [['1110-110', 'pFFpFFF'],
367	      \  ['0010-010', 'pFppFpp'],
368	      \  ['0000-000', 'ppppppp']],
369	      \ [['1111-111', 'FFFFFFF'],
370	      \  ['0011-011', 'FFpFFpp'],
371	      \  ['0000-000', 'ppppppp']]
372	      \ ]
373  for depth in range(5)
374    for u in range(3)
375      unlet! l
376      let l = [0, [1, [2, 3]], {4: 5, 6: {7: 8}}]
377      exe "lockvar " . depth . " l"
378      if u == 1
379        exe "unlockvar l"
380      elseif u == 2
381        exe "unlockvar " . depth . " l"
382      endif
383      let ps = islocked("l").islocked("l[1]").islocked("l[1][1]").islocked("l[1][1][0]").'-'.islocked("l[2]").islocked("l[2]['6']").islocked("l[2]['6'][7]")
384      call assert_equal(expected[depth][u][0], ps, 'depth: ' .. depth)
385      let ps = ''
386      try
387        let l[1][1][0] = 99
388        let ps .= 'p'
389      catch
390        let ps .= 'F'
391      endtry
392      try
393        let l[1][1] = [99]
394        let ps .= 'p'
395      catch
396        let ps .= 'F'
397      endtry
398      try
399        let l[1] = [99]
400        let ps .= 'p'
401      catch
402        let ps .= 'F'
403      endtry
404      try
405        let l[2]['6'][7] = 99
406        let ps .= 'p'
407      catch
408        let ps .= 'F'
409      endtry
410      try
411        let l[2][6] = {99: 99}
412        let ps .= 'p'
413      catch
414        let ps .= 'F'
415      endtry
416      try
417        let l[2] = {99: 99}
418        let ps .= 'p'
419      catch
420        let ps .= 'F'
421      endtry
422      try
423        let l = [99]
424        let ps .= 'p'
425      catch
426        let ps .= 'F'
427      endtry
428      call assert_equal(expected[depth][u][1], ps, 'depth: ' .. depth)
429    endfor
430  endfor
431  call assert_fails("let x=islocked('a b')", 'E488:')
432  let mylist = [1, 2, 3]
433  call assert_fails("let x = islocked('mylist[1:2]')", 'E786:')
434  let mydict = {'k' : 'v'}
435  call assert_fails("let x = islocked('mydict.a')", 'E716:')
436endfunc
437
438" Unletting locked variables
439func Test_list_locked_var_unlet()
440  let expected = [
441	      \ [['1000-000', 'ppppppp'],
442	      \  ['0000-000', 'ppppppp'],
443	      \  ['0000-000', 'ppppppp']],
444	      \ [['1000-000', 'ppFppFp'],
445	      \  ['0000-000', 'ppppppp'],
446	      \  ['0000-000', 'ppppppp']],
447	      \ [['1100-100', 'pFFpFFp'],
448	      \  ['0000-000', 'ppppppp'],
449	      \  ['0000-000', 'ppppppp']],
450	      \ [['1110-110', 'FFFFFFp'],
451	      \  ['0010-010', 'FppFppp'],
452	      \  ['0000-000', 'ppppppp']],
453	      \ [['1111-111', 'FFFFFFp'],
454	      \  ['0011-011', 'FppFppp'],
455	      \  ['0000-000', 'ppppppp']]
456	      \ ]
457
458  for depth in range(5)
459    for u in range(3)
460      unlet! l
461      let l = [0, [1, [2, 3]], {4: 5, 6: {7: 8}}]
462      exe "lockvar " . depth . " l"
463      if u == 1
464        exe "unlockvar l"
465      elseif u == 2
466        exe "unlockvar " . depth . " l"
467      endif
468      let ps = islocked("l").islocked("l[1]").islocked("l[1][1]").islocked("l[1][1][0]").'-'.islocked("l[2]").islocked("l[2]['6']").islocked("l[2]['6'][7]")
469      call assert_equal(expected[depth][u][0], ps, 'depth: ' .. depth)
470      let ps = ''
471      try
472        unlet l[2]['6'][7]
473        let ps .= 'p'
474      catch
475        let ps .= 'F'
476      endtry
477      try
478        unlet l[2][6]
479        let ps .= 'p'
480      catch
481        let ps .= 'F'
482      endtry
483      try
484        unlet l[2]
485        let ps .= 'p'
486      catch
487        let ps .= 'F'
488      endtry
489      try
490        unlet l[1][1][0]
491        let ps .= 'p'
492      catch
493        let ps .= 'F'
494      endtry
495      try
496        unlet l[1][1]
497        let ps .= 'p'
498      catch
499        let ps .= 'F'
500      endtry
501      try
502        unlet l[1]
503        let ps .= 'p'
504      catch
505        let ps .= 'F'
506      endtry
507      try
508        unlet l
509        let ps .= 'p'
510      catch
511        let ps .= 'F'
512      endtry
513      call assert_equal(expected[depth][u][1], ps)
514    endfor
515  endfor
516  " Deleting a list range should fail if the range is locked
517  let l = [1, 2, 3, 4]
518  lockvar l[1:2]
519  call assert_fails('unlet l[1:2]', 'E741:')
520  unlet l
521endfunc
522
523" Locked variables and :unlet or list / dict functions
524
525" No :unlet after lock on dict:
526func Test_dict_lock_unlet()
527  unlet! d
528  let d = {'a': 99, 'b': 100}
529  lockvar 1 d
530  call assert_fails('unlet d.a', 'E741:')
531endfunc
532
533" unlet after lock on dict item
534func Test_dict_item_lock_unlet()
535  unlet! d
536  let d = {'a': 99, 'b': 100}
537  lockvar d.a
538  unlet d.a
539  call assert_equal({'b' : 100}, d)
540endfunc
541
542" filter() after lock on dict item
543func Test_dict_lock_filter()
544  unlet! d
545  let d = {'a': 99, 'b': 100}
546  lockvar d.a
547  call filter(d, 'v:key != "a"')
548  call assert_equal({'b' : 100}, d)
549endfunc
550
551" map() after lock on dict
552func Test_dict_lock_map()
553  unlet! d
554  let d = {'a': 99, 'b': 100}
555  lockvar 1 d
556  call map(d, 'v:val + 200')
557  call assert_equal({'a' : 299, 'b' : 300}, d)
558endfunc
559
560" No extend() after lock on dict item
561func Test_dict_lock_extend()
562  unlet! d
563  let d = {'a': 99, 'b': 100}
564  lockvar d.a
565  call assert_fails("call extend(d, {'a' : 123})", 'E741:')
566  call assert_equal({'a': 99, 'b': 100}, d)
567endfunc
568
569" Cannot use += with a locked dict
570func Test_dict_lock_operator()
571  unlet! d
572  let d = {}
573  lockvar d
574  call assert_fails("let d += {'k' : 10}", 'E741:')
575  unlockvar d
576endfunc
577
578" No remove() of write-protected scope-level variable
579func Tfunc1(this_is_a_long_parameter_name)
580  call assert_fails("call remove(a:, 'this_is_a_long_parameter_name')", 'E742:')
581endfunc
582func Test_dict_scope_var_remove()
583  call Tfunc1('testval')
584endfunc
585
586" No extend() of write-protected scope-level variable
587func Test_dict_scope_var_extend()
588  call assert_fails("call extend(a:, {'this_is_a_long_parameter_name': 1234})", 'E742:')
589endfunc
590
591func Tfunc2(this_is_a_long_parameter_name)
592  call assert_fails("call extend(a:, {'this_is_a_long_parameter_name': 1234})", 'E742:')
593endfunc
594func Test_dict_scope_var_extend_overwrite()
595  call Tfunc2('testval')
596endfunc
597
598" No :unlet of variable in locked scope
599func Test_lock_var_unlet()
600  let b:testvar = 123
601  lockvar 1 b:
602  call assert_fails('unlet b:testvar', 'E741:')
603  unlockvar 1 b:
604  unlet! b:testvar
605endfunc
606
607" No :let += of locked list variable
608func Test_let_lock_list()
609  let l = ['a', 'b', 3]
610  lockvar 1 l
611  call assert_fails("let l += ['x']", 'E741:')
612  call assert_equal(['a', 'b', 3], l)
613
614  unlet l
615  let l = [1, 2, 3, 4]
616  lockvar! l
617  call assert_equal([1, 2, 3, 4], l)
618  unlockvar l[1]
619  call assert_fails('unlet l[0:1]', 'E741:')
620  call assert_equal([1, 2, 3, 4], l)
621  call assert_fails('unlet l[1:2]', 'E741:')
622  call assert_equal([1, 2, 3, 4], l)
623  unlockvar l[1]
624  call assert_fails('let l[0:1] = [0, 1]', 'E741:')
625  call assert_equal([1, 2, 3, 4], l)
626  call assert_fails('let l[1:2] = [0, 1]', 'E741:')
627  call assert_equal([1, 2, 3, 4], l)
628  unlet l
629endfunc
630
631" Locking part of the list
632func Test_let_lock_list_items()
633  let l = [1, 2, 3, 4]
634  lockvar l[2:]
635  call assert_equal(0, islocked('l[0]'))
636  call assert_equal(1, islocked('l[2]'))
637  call assert_equal(1, islocked('l[3]'))
638  call assert_fails('let l[2] = 10', 'E741:')
639  call assert_fails('let l[3] = 20', 'E741:')
640  unlet l
641endfunc
642
643" lockvar/islocked() triggering script autoloading
644func Test_lockvar_script_autoload()
645  let old_rtp = &rtp
646  set rtp+=./sautest
647  lockvar g:footest#x
648  unlockvar g:footest#x
649  call assert_equal(-1, 'g:footest#x'->islocked())
650  call assert_equal(0, exists('g:footest#x'))
651  call assert_equal(1, g:footest#x)
652  let &rtp = old_rtp
653endfunc
654
655" a:000 function argument test
656func s:arg_list_test(...)
657  call assert_fails('let a:000 = [1, 2]', 'E46:')
658  call assert_fails('let a:000[0] = 9', 'E742:')
659  call assert_fails('let a:000[2] = [9, 10]', 'E742:')
660  call assert_fails('let a:000[3] = {9 : 10}', 'E742:')
661
662  " now the tests that should pass
663  let a:000[2][1] = 9
664  call extend(a:000[2], [5, 6])
665  let a:000[3][5] = 8
666  let a:000[3]['a'] = 12
667  call assert_equal([1, 2, [3, 9, 5, 6], {'a': 12, '5': 8}], a:000)
668endfunc
669
670func Test_func_arg_list()
671  call s:arg_list_test(1, 2, [3, 4], {5: 6})
672endfunc
673
674func Test_dict_item_locked()
675endfunc
676
677" Tests for reverse(), sort(), uniq()
678func Test_reverse_sort_uniq()
679  let l = ['-0', 'A11', 2, 2, 'xaaa', 4, 'foo', 'foo6', 'foo', [0, 1, 2], 'x8', [0, 1, 2], 1.5]
680  call assert_equal(['-0', 'A11', 2, 'xaaa', 4, 'foo', 'foo6', 'foo', [0, 1, 2], 'x8', [0, 1, 2], 1.5], uniq(copy(l)))
681  call assert_equal([1.5, [0, 1, 2], 'x8', [0, 1, 2], 'foo', 'foo6', 'foo', 4, 'xaaa', 2, 2, 'A11', '-0'], reverse(l))
682  call assert_equal([1.5, [0, 1, 2], 'x8', [0, 1, 2], 'foo', 'foo6', 'foo', 4, 'xaaa', 2, 2, 'A11', '-0'], reverse(reverse(l)))
683  if has('float')
684    call assert_equal(['-0', 'A11', 'foo', 'foo', 'foo6', 'x8', 'xaaa', 1.5, 2, 2, 4, [0, 1, 2], [0, 1, 2]], sort(l))
685    call assert_equal([[0, 1, 2], [0, 1, 2], 4, 2, 2, 1.5, 'xaaa', 'x8', 'foo6', 'foo', 'foo', 'A11', '-0'], reverse(sort(l)))
686    call assert_equal(['-0', 'A11', 'foo', 'foo', 'foo6', 'x8', 'xaaa', 1.5, 2, 2, 4, [0, 1, 2], [0, 1, 2]], sort(reverse(sort(l))))
687    call assert_equal(['-0', 'A11', 'foo', 'foo6', 'x8', 'xaaa', 1.5, 2, 4, [0, 1, 2]], uniq(sort(l)))
688
689    let l = [7, 9, 'one', 18, 12, 22, 'two', 10.0e-16, -1, 'three', 0xff, 0.22, 'four']
690    call assert_equal([-1, 'one', 'two', 'three', 'four', 1.0e-15, 0.22, 7, 9, 12, 18, 22, 255], sort(copy(l), 'n'))
691
692    let l = [7, 9, 18, 12, 22, 10.0e-16, -1, 0xff, 0, -0, 0.22, 'bar', 'BAR', 'Bar', 'Foo', 'FOO', 'foo', 'FOOBAR', {}, []]
693    call assert_equal(['bar', 'BAR', 'Bar', 'Foo', 'FOO', 'foo', 'FOOBAR', -1, 0, 0, 0.22, 1.0e-15, 12, 18, 22, 255, 7, 9, [], {}], sort(copy(l), 1))
694    call assert_equal(['bar', 'BAR', 'Bar', 'Foo', 'FOO', 'foo', 'FOOBAR', -1, 0, 0, 0.22, 1.0e-15, 12, 18, 22, 255, 7, 9, [], {}], sort(copy(l), 'i'))
695    call assert_equal(['BAR', 'Bar', 'FOO', 'FOOBAR', 'Foo', 'bar', 'foo', -1, 0, 0, 0.22, 1.0e-15, 12, 18, 22, 255, 7, 9, [], {}], sort(copy(l)))
696  endif
697
698  call assert_fails('call reverse("")', 'E899:')
699  call assert_fails('call uniq([1, 2], {x, y -> []})', 'E745:')
700  call assert_fails("call sort([1, 2], function('min'), 1)", "E715:")
701  call assert_fails("call sort([1, 2], function('invalid_func'))", "E700:")
702  call assert_fails("call sort([1, 2], function('min'))", "E118:")
703endfunc
704
705" reduce a list or a blob
706func Test_reduce()
707  call assert_equal(1, reduce([], { acc, val -> acc + val }, 1))
708  call assert_equal(10, reduce([1, 3, 5], { acc, val -> acc + val }, 1))
709  call assert_equal(2 * (2 * ((2 * 1) + 2) + 3) + 4, reduce([2, 3, 4], { acc, val -> 2 * acc + val }, 1))
710  call assert_equal('a x y z', ['x', 'y', 'z']->reduce({ acc, val -> acc .. ' ' .. val}, 'a'))
711  call assert_equal(#{ x: 1, y: 1, z: 1 }, ['x', 'y', 'z']->reduce({ acc, val -> extend(acc, { val: 1 }) }, {}))
712  call assert_equal([0, 1, 2, 3], reduce([1, 2, 3], function('add'), [0]))
713
714  let l = ['x', 'y', 'z']
715  call assert_equal(42, reduce(l, function('get'), #{ x: #{ y: #{ z: 42 } } }))
716  call assert_equal(['x', 'y', 'z'], l)
717
718  call assert_equal(1, reduce([1], { acc, val -> acc + val }))
719  call assert_equal('x y z', reduce(['x', 'y', 'z'], { acc, val -> acc .. ' ' .. val }))
720  call assert_equal(120, range(1, 5)->reduce({ acc, val -> acc * val }))
721  call assert_fails("call reduce([], { acc, val -> acc + val })", 'E998: Reduce of an empty List with no initial value')
722
723  call assert_equal(1, reduce(0z, { acc, val -> acc + val }, 1))
724  call assert_equal(1 + 0xaf + 0xbf + 0xcf, reduce(0zAFBFCF, { acc, val -> acc + val }, 1))
725  call assert_equal(2 * (2 * 1 + 0xaf) + 0xbf, 0zAFBF->reduce({ acc, val -> 2 * acc + val }, 1))
726
727  call assert_equal(0xff, reduce(0zff, { acc, val -> acc + val }))
728  call assert_equal(2 * (2 * 0xaf + 0xbf) + 0xcf, reduce(0zAFBFCF, { acc, val -> 2 * acc + val }))
729  call assert_fails("call reduce(0z, { acc, val -> acc + val })", 'E998: Reduce of an empty Blob with no initial value')
730
731  call assert_fails("call reduce({}, { acc, val -> acc + val }, 1)", 'E897:')
732  call assert_fails("call reduce(0, { acc, val -> acc + val }, 1)", 'E897:')
733  call assert_fails("call reduce('', { acc, val -> acc + val }, 1)", 'E897:')
734  call assert_fails("call reduce([1, 2], 'Xdoes_not_exist')", 'E117:')
735  call assert_fails("echo reduce(0z01, { acc, val -> 2 * acc + val }, '')", 'E39:')
736
737  let g:lut = [1, 2, 3, 4]
738  func EvilRemove()
739    call remove(g:lut, 1)
740    return 1
741  endfunc
742  call assert_fails("call reduce(g:lut, { acc, val -> EvilRemove() }, 1)", 'E742:')
743  unlet g:lut
744  delfunc EvilRemove
745
746  call assert_equal(42, reduce(test_null_list(), function('add'), 42))
747  call assert_equal(42, reduce(test_null_blob(), function('add'), 42))
748
749  " should not crash
750  call assert_fails('echo reduce([1], test_null_function())', 'E1132:')
751  call assert_fails('echo reduce([1], test_null_partial())', 'E1132:')
752endfunc
753
754" splitting a string to a List using split()
755func Test_str_split()
756  call assert_equal(['aa', 'bb'], split('  aa  bb '))
757  call assert_equal(['aa', 'bb'], split('  aa  bb  ', '\W\+', 0))
758  call assert_equal(['', 'aa', 'bb', ''], split('  aa  bb  ', '\W\+', 1))
759  call assert_equal(['', '', 'aa', '', 'bb', '', ''], split('  aa  bb  ', '\W', 1))
760  call assert_equal(['aa', '', 'bb'], split(':aa::bb:', ':', 0))
761  call assert_equal(['', 'aa', '', 'bb', ''], split(':aa::bb:', ':', 1))
762  call assert_equal(['aa', '', 'bb', 'cc', ''], split('aa,,bb, cc,', ',\s*', 1))
763  call assert_equal(['a', 'b', 'c'], split('abc', '\zs'))
764  call assert_equal(['', 'a', '', 'b', '', 'c', ''], split('abc', '\zs', 1))
765  call assert_fails("call split('abc', [])", 'E730:')
766  call assert_fails("call split('abc', 'b', [])", 'E745:')
767  call assert_equal(['abc'], split('abc', '\\%('))
768endfunc
769
770" compare recursively linked list and dict
771func Test_listdict_compare()
772  let l = [1, 2, 3, 4]
773  let d = {'1': 1, '2': l, '3': 3}
774  let l[1] = d
775  call assert_true(l == l)
776  call assert_true(d == d)
777  call assert_false(l != deepcopy(l))
778  call assert_false(d != deepcopy(d))
779
780  " comparison errors
781  call assert_fails('echo [1, 2] =~ {}', 'E691:')
782  call assert_fails('echo [1, 2] =~ [1, 2]', 'E692:')
783  call assert_fails('echo {} =~ 5', 'E735:')
784  call assert_fails('echo {} =~ {}', 'E736:')
785endfunc
786
787  " compare complex recursively linked list and dict
788func Test_listdict_compare_complex()
789  let l = []
790  call add(l, l)
791  let dict4 = {"l": l}
792  call add(dict4.l, dict4)
793  let lcopy = deepcopy(l)
794  let dict4copy = deepcopy(dict4)
795  call assert_true(l == lcopy)
796  call assert_true(dict4 == dict4copy)
797endfunc
798
799" Test for extending lists and dictionaries
800func Test_listdict_extend()
801  " Test extend() with lists
802
803  " Pass the same List to extend()
804  let l = [1, 2, 3]
805  call assert_equal([1, 2, 3, 1, 2, 3], extend(l, l))
806  call assert_equal([1, 2, 3, 1, 2, 3], l)
807
808  let l = [1, 2, 3]
809  call assert_equal([1, 2, 3, 4, 5, 6], extend(l, [4, 5, 6]))
810  call assert_equal([1, 2, 3, 4, 5, 6], l)
811
812  let l = [1, 2, 3]
813  call extend(l, [4, 5, 6], 0)
814  call assert_equal([4, 5, 6, 1, 2, 3], l)
815
816  let l = [1, 2, 3]
817  call extend(l, [4, 5, 6], 1)
818  call assert_equal([1, 4, 5, 6, 2, 3], l)
819
820  let l = [1, 2, 3]
821  call extend(l, [4, 5, 6], 3)
822  call assert_equal([1, 2, 3, 4, 5, 6], l)
823
824  let l = [1, 2, 3]
825  call extend(l, [4, 5, 6], -1)
826  call assert_equal([1, 2, 4, 5, 6, 3], l)
827
828  let l = [1, 2, 3]
829  call extend(l, [4, 5, 6], -3)
830  call assert_equal([4, 5, 6, 1, 2,  3], l)
831
832  let l = [1, 2, 3]
833  call assert_fails("call extend(l, [4, 5, 6], 4)", 'E684:')
834  call assert_fails("call extend(l, [4, 5, 6], -4)", 'E684:')
835  if has('float')
836    call assert_fails("call extend(l, [4, 5, 6], 1.2)", 'E805:')
837  endif
838
839  " Test extend() with dictionaries.
840
841  " Pass the same Dict to extend()
842  let d = { 'a': {'b': 'B'}}
843  call extend(d, d)
844  call assert_equal({'a': {'b': 'B'}}, d)
845
846  let d = {'a': 'A', 'b': 'B'}
847  call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, extend(d, {'b': 0, 'c':'C'}))
848  call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, d)
849
850  let d = {'a': 'A', 'b': 'B'}
851  call extend(d, {'a': 'A', 'b': 0, 'c': 'C'}, "force")
852  call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, d)
853
854  let d = {'a': 'A', 'b': 'B'}
855  call extend(d, {'b': 0, 'c':'C'}, "keep")
856  call assert_equal({'a': 'A', 'b': 'B', 'c': 'C'}, d)
857
858  let d = {'a': 'A', 'b': 'B'}
859  call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 'error')", 'E737:')
860  call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 'xxx')", 'E475:')
861  if has('float')
862    call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 1.2)", 'E475:')
863  endif
864  call assert_equal({'a': 'A', 'b': 'B'}, d)
865
866  call assert_fails("call extend([1, 2], 1)", 'E712:')
867  call assert_fails("call extend([1, 2], {})", 'E712:')
868
869  " Extend g: dictionary with an invalid variable name
870  call assert_fails("call extend(g:, {'-!' : 10})", 'E461:')
871
872  " Extend a list with itself.
873  let l = [1, 5, 7]
874  call extend(l, l, 0)
875  call assert_equal([1, 5, 7, 1, 5, 7], l)
876  let l = [1, 5, 7]
877  call extend(l, l, 1)
878  call assert_equal([1, 1, 5, 7, 5, 7], l)
879  let l = [1, 5, 7]
880  call extend(l, l, 2)
881  call assert_equal([1, 5, 1, 5, 7, 7], l)
882  let l = [1, 5, 7]
883  call extend(l, l, 3)
884  call assert_equal([1, 5, 7, 1, 5, 7], l)
885endfunc
886
887func Test_listdict_extendnew()
888  " Test extendnew() with lists
889  let l = [1, 2, 3]
890  call assert_equal([1, 2, 3, 4, 5], extendnew(l, [4, 5]))
891  call assert_equal([1, 2, 3], l)
892
893  " Test extend() with dictionaries.
894  let d = {'a': {'b': 'B'}}
895  call assert_equal({'a': {'b': 'B'}, 'c': 'cc'}, extendnew(d, {'c': 'cc'}))
896  call assert_equal({'a': {'b': 'B'}}, d)
897endfunc
898
899func s:check_scope_dict(x, fixed)
900  func s:gen_cmd(cmd, x)
901    return substitute(a:cmd, '\<x\ze:', a:x, 'g')
902  endfunc
903
904  let cmd = s:gen_cmd('let x:foo = 1', a:x)
905  if a:fixed
906    call assert_fails(cmd, 'E461:')
907  else
908    exe cmd
909    exe s:gen_cmd('call assert_equal(1, x:foo)', a:x)
910  endif
911
912  let cmd = s:gen_cmd('let x:["bar"] = 2', a:x)
913  if a:fixed
914    call assert_fails(cmd, 'E461:')
915  else
916    exe cmd
917    exe s:gen_cmd('call assert_equal(2, x:bar)', a:x)
918  endif
919
920  let cmd = s:gen_cmd('call extend(x:, {"baz": 3})', a:x)
921  if a:fixed
922    call assert_fails(cmd, 'E742:')
923  else
924    exe cmd
925    exe s:gen_cmd('call assert_equal(3, x:baz)', a:x)
926  endif
927
928  if a:fixed
929    if a:x ==# 'a'
930      call assert_fails('unlet a:x', 'E795:')
931      call assert_fails('call remove(a:, "x")', 'E742:')
932    elseif a:x ==# 'v'
933      call assert_fails('unlet v:count', 'E795:')
934      call assert_fails('call remove(v:, "count")', 'E742:')
935    endif
936  else
937    exe s:gen_cmd('unlet x:foo', a:x)
938    exe s:gen_cmd('unlet x:bar', a:x)
939    exe s:gen_cmd('call remove(x:, "baz")', a:x)
940  endif
941
942  delfunc s:gen_cmd
943endfunc
944
945func Test_scope_dict()
946  " Test for g:
947  call s:check_scope_dict('g', v:false)
948
949  " Test for s:
950  call s:check_scope_dict('s', v:false)
951
952  " Test for l:
953  call s:check_scope_dict('l', v:false)
954
955  " Test for a:
956  call s:check_scope_dict('a', v:true)
957
958  " Test for b:
959  call s:check_scope_dict('b', v:false)
960
961  " Test for w:
962  call s:check_scope_dict('w', v:false)
963
964  " Test for t:
965  call s:check_scope_dict('t', v:false)
966
967  " Test for v:
968  call s:check_scope_dict('v', v:true)
969endfunc
970
971" Test for deep nesting of lists (> 100)
972func Test_deep_nested_list()
973  let deep_list = []
974  let l = deep_list
975  for i in range(102)
976    let newlist = []
977    call add(l, newlist)
978    let l = newlist
979  endfor
980  call add(l, 102)
981
982  call assert_fails('let m = deepcopy(deep_list)', 'E698:')
983  call assert_fails('lockvar 110 deep_list', 'E743:')
984  call assert_fails('unlockvar 110 deep_list', 'E743:')
985  call assert_fails('let x = execute("echo deep_list")', 'E724:')
986  call test_garbagecollect_now()
987  unlet deep_list
988endfunc
989
990" Test for deep nesting of dicts (> 100)
991func Test_deep_nested_dict()
992  let deep_dict = {}
993  let d = deep_dict
994  for i in range(102)
995    let newdict = {}
996    let d.k = newdict
997    let d = newdict
998  endfor
999  let d.k = 'v'
1000
1001  call assert_fails('let m = deepcopy(deep_dict)', 'E698:')
1002  call assert_fails('lockvar 110 deep_dict', 'E743:')
1003  call assert_fails('unlockvar 110 deep_dict', 'E743:')
1004  call assert_fails('let x = execute("echo deep_dict")', 'E724:')
1005  call test_garbagecollect_now()
1006  unlet deep_dict
1007endfunc
1008
1009" List and dict indexing tests
1010func Test_listdict_index()
1011  call assert_fails('echo function("min")[0]', 'E695:')
1012  call assert_fails('echo v:true[0]', 'E909:')
1013  let d = {'k' : 10}
1014  call assert_fails('echo d.', 'E15:')
1015  call assert_fails('echo d[1:2]', 'E719:')
1016  call assert_fails("let v = [4, 6][{-> 1}]", 'E729:')
1017  call assert_fails("let v = range(5)[2:[]]", 'E730:')
1018  call assert_fails("let v = range(5)[2:{-> 2}(]", ['E15:', 'E116:'])
1019  call assert_fails("let v = range(5)[2:3", 'E111:')
1020  call assert_fails("let l = insert([1,2,3], 4, 10)", 'E684:')
1021  call assert_fails("let l = insert([1,2,3], 4, -10)", 'E684:')
1022  call assert_fails("let l = insert([1,2,3], 4, [])", 'E745:')
1023  let l = [1, 2, 3]
1024  call assert_fails("let l[i] = 3", 'E121:')
1025  call assert_fails("let l[1.1] = 4", 'E805:')
1026  call assert_fails("let l[:i] = [4, 5]", 'E121:')
1027  call assert_fails("let l[:3.2] = [4, 5]", 'E805:')
1028  let t = test_unknown()
1029  call assert_fails("echo t[0]", 'E685:')
1030endfunc
1031
1032" Test for a null list
1033func Test_null_list()
1034  let l = test_null_list()
1035  call assert_equal(0, join(test_null_list()))
1036  call assert_equal('', join(l))
1037  call assert_equal(0, len(l))
1038  call assert_equal(1, empty(l))
1039  call assert_fails('let s = join([1, 2], [])', 'E730:')
1040  call assert_equal([], split(test_null_string()))
1041  call assert_equal([], l[:2])
1042  call assert_true([] == l)
1043  call assert_equal('[]', string(l))
1044  call assert_equal(0, sort(test_null_list()))
1045  call assert_equal([], sort(l))
1046  call assert_equal(0, uniq(test_null_list()))
1047  call assert_equal([], uniq(l))
1048  let k = [] + l
1049  call assert_equal([], k)
1050  let k = l + []
1051  call assert_equal([], k)
1052  call assert_equal(0, len(copy(l)))
1053  call assert_equal(0, count(l, 5))
1054  call assert_equal([], deepcopy(l))
1055  call assert_equal(5, get(l, 2, 5))
1056  call assert_equal(-1, index(l, 2, 5))
1057  call assert_equal(0, insert(test_null_list(), 2, -1))
1058  call assert_fails('call insert(l, 2, -1)', 'E684:')
1059  call assert_equal(0, min(l))
1060  call assert_equal(0, max(l))
1061  call assert_equal(0, remove(test_null_list(), 0, 2))
1062  call assert_fails('call remove(l, 0, 2)', 'E684:')
1063  call assert_equal([], repeat(l, 2))
1064  call assert_equal(0, reverse(test_null_list()))
1065  call assert_equal([], reverse(l))
1066  call assert_equal(0, sort(test_null_list()))
1067  call assert_equal([], sort(l))
1068  call assert_equal('[]', string(l))
1069  call assert_fails('call extend(test_null_list(), test_null_list())', 'E1134:')
1070  call assert_equal([], extend(l, l, 0))
1071  lockvar l
1072  call assert_equal(1, islocked('l'))
1073  unlockvar l
1074endfunc
1075
1076" Test for a null dict
1077func Test_null_dict()
1078  call assert_equal(test_null_dict(), test_null_dict())
1079  let d = test_null_dict()
1080  call assert_equal({}, d)
1081  call assert_equal(0, len(d))
1082  call assert_equal(1, empty(d))
1083  call assert_equal(0, items(test_null_dict()))
1084  call assert_equal([], items(d))
1085  call assert_equal(0, keys(test_null_dict()))
1086  call assert_equal([], keys(d))
1087  call assert_equal(0, values(test_null_dict()))
1088  call assert_equal([], values(d))
1089  call assert_false(has_key(d, 'k'))
1090  call assert_equal('{}', string(d))
1091  call assert_fails('let x = d[10]', 'E716:')
1092  call assert_equal({}, {})
1093  call assert_equal(0, len(copy(d)))
1094  call assert_equal(0, count(d, 'k'))
1095  call assert_equal({}, deepcopy(d))
1096  call assert_equal(20, get(d, 'k', 20))
1097  call assert_equal(0, min(d))
1098  call assert_equal(0, max(d))
1099  call assert_equal(0, remove(test_null_dict(), 'k'))
1100  call assert_fails("call remove(d, 'k')", 'E716:')
1101  call assert_equal('{}', string(d))
1102  call assert_fails('call extend(test_null_dict(), test_null_dict())', 'E1133:')
1103  call assert_equal({}, extend(d, d, 'keep'))
1104  lockvar d
1105  call assert_equal(1, islocked('d'))
1106  unlockvar d
1107endfunc
1108
1109" vim: shiftwidth=2 sts=2 expandtab
1110