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