blob: 32e376899b0b75451f410eddff6521fe42f788f3 [file] [log] [blame]
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001" Tests for the List and Dict types
2
Bram Moolenaar63cb6562021-07-20 22:21:59 +02003source vim9.vim
4
Bram Moolenaarfb094e12017-11-05 20:59:28 +01005func TearDown()
6 " Run garbage collection after every test
7 call test_garbagecollect_now()
8endfunc
9
10" Tests for List type
11
12" List creation
13func Test_list_create()
14 " Creating List directly with different types
15 let l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
16 call assert_equal("[1, 'as''d', [1, 2, function('strlen')], {'a': 1}]", string(l))
17 call assert_equal({'a' : 1}, l[-1])
18 call assert_equal(1, l[-4])
19 let x = 10
20 try
21 let x = l[-5]
22 catch
23 call assert_match('E684:', v:exception)
24 endtry
25 call assert_equal(10, x)
26endfunc
27
Bram Moolenaar4d4d1cd2020-07-30 22:14:33 +020028" This was allowed in legacy Vim script
29let s:list_with_spaces = [1 , 2 , 3]
30
Bram Moolenaarfb094e12017-11-05 20:59:28 +010031" List slices
32func Test_list_slice()
33 let l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
34 call assert_equal([1, 'as''d', [1, 2, function('strlen')], {'a': 1}], l[:])
35 call assert_equal(['as''d', [1, 2, function('strlen')], {'a': 1}], l[1:])
36 call assert_equal([1, 'as''d', [1, 2, function('strlen')]], l[:-2])
37 call assert_equal([1, 'as''d', [1, 2, function('strlen')], {'a': 1}], l[0:8])
38 call assert_equal([], l[8:-1])
Bram Moolenaar8b633132020-03-20 18:20:51 +010039 call assert_equal([], l[0:-10])
Bram Moolenaarea04a6e2020-04-23 13:38:02 +020040 " perform an operation on a list slice
41 let l = [1, 2, 3]
42 let l[:1] += [1, 2]
43 let l[2:] -= [1]
44 call assert_equal([2, 4, 2], l)
Bram Moolenaarfb094e12017-11-05 20:59:28 +010045endfunc
46
47" List identity
48func Test_list_identity()
Bram Moolenaar63cb6562021-07-20 22:21:59 +020049 let lines =<< trim END
50 VAR l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
51 VAR ll = l
52 VAR lx = copy(l)
53 call assert_true(l == ll)
54 call assert_false(l isnot ll)
55 call assert_true(l is ll)
56 call assert_true(l == lx)
57 call assert_false(l is lx)
58 call assert_true(l isnot lx)
59 END
60 call CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +010061endfunc
62
63" removing items with :unlet
64func Test_list_unlet()
Bram Moolenaar63cb6562021-07-20 22:21:59 +020065 let lines =<< trim END
66 VAR l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
67 unlet l[2]
68 call assert_equal([1, 'as''d', {'a': 1}], l)
69 LET l = range(8)
70 unlet l[: 3]
71 unlet l[1 :]
72 call assert_equal([4], l)
Bram Moolenaarfb094e12017-11-05 20:59:28 +010073
Bram Moolenaar63cb6562021-07-20 22:21:59 +020074 #" removing items out of range: silently skip items that don't exist
75 LET l = [0, 1, 2, 3]
76 unlet l[2 : 2]
77 call assert_equal([0, 1, 3], l)
78 LET l = [0, 1, 2, 3]
79 unlet l[2 : 3]
80 call assert_equal([0, 1], l)
81 LET l = [0, 1, 2, 3]
82 unlet l[2 : 4]
83 call assert_equal([0, 1], l)
84 LET l = [0, 1, 2, 3]
85 unlet l[2 : 5]
86 call assert_equal([0, 1], l)
87 LET l = [0, 1, 2, 3]
88 unlet l[-2 : 2]
89 call assert_equal([0, 1, 3], l)
90 LET l = [0, 1, 2, 3]
91 unlet l[-3 : 2]
92 call assert_equal([0, 3], l)
93 LET l = [0, 1, 2, 3]
94 unlet l[-4 : 2]
95 call assert_equal([3], l)
96 LET l = [0, 1, 2, 3]
97 unlet l[-5 : 2]
98 call assert_equal([3], l)
99 LET l = [0, 1, 2, 3]
100 unlet l[-6 : 2]
101 call assert_equal([3], l)
102 END
103 call CheckLegacyAndVim9Success(lines)
104
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100105 let l = [0, 1, 2, 3]
106 unlet l[2:2]
107 call assert_equal([0, 1, 3], l)
108 let l = [0, 1, 2, 3]
109 unlet l[2:3]
110 call assert_equal([0, 1], l)
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200111
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100112 let l = [0, 1, 2, 3]
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200113 call assert_fails('unlet l[2:1]', 'E684:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100114 let l = [0, 1, 2, 3]
Bram Moolenaare2e40752020-09-04 21:18:46 +0200115 call assert_fails('unlet l[-1:2]', 'E684:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100116endfunc
117
118" assignment to a list
119func Test_list_assign()
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200120 let lines =<< trim END
121 VAR l = [0, 1, 2, 3]
122 VAR va = 0
123 VAR vb = 0
124 LET [va, vb] = l[2 : 3]
125 call assert_equal([2, 3], [va, vb])
126 END
127 call CheckLegacyAndVim9Success(lines)
128
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100129 let l = [0, 1, 2, 3]
Bram Moolenaare2e40752020-09-04 21:18:46 +0200130 call assert_fails('let [va, vb] = l', 'E687:')
131 call assert_fails('let [va, vb] = l[1:1]', 'E688:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100132endfunc
133
134" test for range assign
135func Test_list_range_assign()
136 let l = [0]
137 let l[:] = [1, 2]
138 call assert_equal([1, 2], l)
Bram Moolenaar8b633132020-03-20 18:20:51 +0100139 let l[-4:-1] = [5, 6]
140 call assert_equal([5, 6], l)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100141endfunc
142
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200143" Test removing items in list
144func Test_list_func_remove()
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200145 let lines =<< trim END
146 #" Test removing 1 element
147 VAR l = [1, 2, 3, 4]
148 call assert_equal(1, remove(l, 0))
149 call assert_equal([2, 3, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200150
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200151 LET l = [1, 2, 3, 4]
152 call assert_equal(2, remove(l, 1))
153 call assert_equal([1, 3, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200154
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200155 LET l = [1, 2, 3, 4]
156 call assert_equal(4, remove(l, -1))
157 call assert_equal([1, 2, 3], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200158
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200159 #" Test removing range of element(s)
160 LET l = [1, 2, 3, 4]
161 call assert_equal([3], remove(l, 2, 2))
162 call assert_equal([1, 2, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200163
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200164 LET l = [1, 2, 3, 4]
165 call assert_equal([2, 3], remove(l, 1, 2))
166 call assert_equal([1, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200167
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200168 LET l = [1, 2, 3, 4]
169 call assert_equal([2, 3], remove(l, -3, -2))
170 call assert_equal([1, 4], l)
171 END
172 call CheckLegacyAndVim9Success(lines)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200173
174 " Test invalid cases
175 let l = [1, 2, 3, 4]
176 call assert_fails("call remove(l, 5)", 'E684:')
177 call assert_fails("call remove(l, 1, 5)", 'E684:')
178 call assert_fails("call remove(l, 3, 2)", 'E16:')
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +0100179 call assert_fails("call remove(1, 0)", 'E896:')
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200180 call assert_fails("call remove(l, l)", 'E745:')
181endfunc
182
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200183" List add() function
184func Test_list_add()
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200185 let lines =<< trim END
186 VAR l = []
187 call add(l, 1)
188 call add(l, [2, 3])
189 call add(l, [])
190 call add(l, test_null_list())
191 call add(l, {'k': 3})
192 call add(l, {})
193 call add(l, test_null_dict())
194 call assert_equal([1, [2, 3], [], [], {'k': 3}, {}, {}], l)
195 END
196 call CheckLegacyAndVim9Success(lines)
197
198 " weird legacy behavior
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200199 call assert_equal(1, add(test_null_list(), 4))
200endfunc
201
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100202" Tests for Dictionary type
203
204func Test_dict()
205 " Creating Dictionary directly with different types
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200206 let lines =<< trim END
207 VAR d = {'1': 'asd', 'b': [1, 2, function('strlen')], '-1': {'a': 1}, }
208 call assert_equal("{'1': 'asd', 'b': [1, 2, function('strlen')], '-1': {'a': 1}}", string(d))
209 call assert_equal('asd', d.1)
210 call assert_equal(['-1', '1', 'b'], sort(keys(d)))
211 call assert_equal(['asd', [1, 2, function('strlen')], {'a': 1}], values(d))
212 call extend(d, {3: 33, 1: 99})
213 call extend(d, {'b': 'bbb', 'c': 'ccc'}, "keep")
214 call assert_equal({'c': 'ccc', '1': 99, 'b': [1, 2, function('strlen')], '3': 33, '-1': {'a': 1}}, d)
215 END
216 call CheckLegacyAndVim9Success(lines)
217
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100218 let d = {001: 'asd', 'b': [1, 2, function('strlen')], -1: {'a': 1},}
219 call assert_equal("{'1': 'asd', 'b': [1, 2, function('strlen')], '-1': {'a': 1}}", string(d))
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200220
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100221 let v = []
222 for [key, val] in items(d)
223 call extend(v, [key, val])
224 unlet key val
225 endfor
226 call assert_equal(['1','asd','b',[1, 2, function('strlen')],'-1',{'a': 1}], v)
227
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200228 call extend(d, {3: 33, 1: 99})
Bram Moolenaare2e40752020-09-04 21:18:46 +0200229 call assert_fails("call extend(d, {3:333,4:444}, 'error')", 'E737:')
Bram Moolenaar08f41572020-04-20 16:50:00 +0200230
231 " duplicate key
232 call assert_fails("let d = {'k' : 10, 'k' : 20}", 'E721:')
233 " missing comma
234 call assert_fails("let d = {'k' : 10 'k' : 20}", 'E722:')
235 " missing curly brace
236 call assert_fails("let d = {'k' : 10,", 'E723:')
237 " invalid key
238 call assert_fails('let d = #{++ : 10}', 'E15:')
239 " wrong type for key
240 call assert_fails('let d={[] : 10}', 'E730:')
241 " undefined variable as value
242 call assert_fails("let d={'k' : i}", 'E121:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100243endfunc
244
Bram Moolenaar4d4d1cd2020-07-30 22:14:33 +0200245" This was allowed in legacy Vim script
246let s:dict_with_spaces = {'one' : 1 , 'two' : 2 , 'three' : 3}
247let s:dict_with_spaces_lit = #{one : 1 , two : 2 , three : 3}
248
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100249" Dictionary identity
250func Test_dict_identity()
251 let d = {001: 'asd', 'b': [1, 2, function('strlen')], -1: {'a': 1},}
252 let dd = d
253 let dx = copy(d)
254 call assert_true(d == dd)
255 call assert_false(d isnot dd)
256 call assert_true(d is dd)
257 call assert_true(d == dx)
258 call assert_false(d is dx)
259 call assert_true(d isnot dx)
260endfunc
261
262" removing items with :unlet
263func Test_dict_unlet()
264 let d = {'b':'bbb', '1': 99, '3': 33, '-1': {'a': 1}}
265 unlet d.b
266 unlet d[-1]
267 call assert_equal({'1': 99, '3': 33}, d)
268endfunc
269
270" manipulating a big Dictionary (hashtable.c has a border of 1000 entries)
271func Test_dict_big()
272 let d = {}
273 for i in range(1500)
274 let d[i] = 3000 - i
275 endfor
276 call assert_equal([3000, 2900, 2001, 1600, 1501], [d[0], d[100], d[999], d[1400], d[1499]])
277 let str = ''
278 try
279 let n = d[1500]
280 catch
Bram Moolenaar6d967122020-10-30 19:06:18 +0100281 let str = substitute(v:exception, '\v(.{14}).*( "\d{4}").*', '\1\2', '')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100282 endtry
Bram Moolenaar6d967122020-10-30 19:06:18 +0100283 call assert_equal('Vim(let):E716: "1500"', str)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100284
285 " lookup each items
286 for i in range(1500)
287 call assert_equal(3000 - i, d[i])
288 endfor
289 let i += 1
290
291 " delete even items
292 while i >= 2
293 let i -= 2
294 unlet d[i]
295 endwhile
296 call assert_equal('NONE', get(d, 1500 - 100, 'NONE'))
297 call assert_equal(2999, d[1])
298
299 " delete odd items, checking value, one intentionally wrong
300 let d[33] = 999
301 let i = 1
302 while i < 1500
303 if i != 33
304 call assert_equal(3000 - i, d[i])
305 else
306 call assert_equal(999, d[i])
307 endif
308 unlet d[i]
309 let i += 2
310 endwhile
311 call assert_equal({}, d)
312 unlet d
313endfunc
314
315" Dictionary function
316func Test_dict_func()
317 let d = {}
318 func d.func(a) dict
319 return a:a . len(self.data)
320 endfunc
321 let d.data = [1,2,3]
322 call assert_equal('len: 3', d.func('len: '))
323 let x = d.func('again: ')
324 call assert_equal('again: 3', x)
325 let Fn = d.func
326 call assert_equal('xxx3', Fn('xxx'))
327endfunc
328
Bram Moolenaarb13ab992020-07-27 21:43:28 +0200329func Test_dict_assign()
330 let d = {}
331 let d.1 = 1
332 let d._ = 2
333 call assert_equal({'1': 1, '_': 2}, d)
Bram Moolenaar3a3b10e2021-06-26 15:00:59 +0200334
335 let n = 0
336 call assert_fails('let n.key = 3', 'E1203: Dot can only be used on a dictionary: n.key = 3')
Bram Moolenaarb13ab992020-07-27 21:43:28 +0200337endfunc
338
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100339" Function in script-local List or Dict
340func Test_script_local_dict_func()
341 let g:dict = {}
342 function g:dict.func() dict
343 return 'g:dict.func' . self.foo[1] . self.foo[0]('asdf')
344 endfunc
345 let g:dict.foo = ['-', 2, 3]
346 call insert(g:dict.foo, function('strlen'))
347 call assert_equal('g:dict.func-4', g:dict.func())
348 unlet g:dict
349endfunc
350
Bram Moolenaar08f41572020-04-20 16:50:00 +0200351" Test removing items in a dictionary
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200352func Test_dict_func_remove()
353 let d = {1:'a', 2:'b', 3:'c'}
354 call assert_equal('b', remove(d, 2))
355 call assert_equal({1:'a', 3:'c'}, d)
356
357 call assert_fails("call remove(d, 1, 2)", 'E118:')
358 call assert_fails("call remove(d, 'a')", 'E716:')
359 call assert_fails("call remove(d, [])", 'E730:')
360endfunc
361
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100362" Nasty: remove func from Dict that's being called (works)
363func Test_dict_func_remove_in_use()
364 let d = {1:1}
365 func d.func(a)
366 return "a:" . a:a
367 endfunc
368 let expected = 'a:' . string(get(d, 'func'))
369 call assert_equal(expected, d.func(string(remove(d, 'func'))))
370endfunc
371
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200372func Test_dict_literal_keys()
Bram Moolenaar4c6d9042019-07-16 22:04:02 +0200373 call assert_equal({'one': 1, 'two2': 2, '3three': 3, '44': 4}, #{one: 1, two2: 2, 3three: 3, 44: 4},)
Bram Moolenaarb8be54d2019-07-14 18:22:59 +0200374
375 " why *{} cannot be used
376 let blue = 'blue'
377 call assert_equal('6', trim(execute('echo 2 *{blue: 3}.blue')))
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200378endfunc
379
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100380" Nasty: deepcopy() dict that refers to itself (fails when noref used)
381func Test_dict_deepcopy()
382 let d = {1:1, 2:2}
383 let l = [4, d, 6]
384 let d[3] = l
385 let dc = deepcopy(d)
Bram Moolenaare2e40752020-09-04 21:18:46 +0200386 call assert_fails('call deepcopy(d, 1)', 'E698:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100387 let l2 = [0, l, l, 3]
388 let l[1] = l2
389 let l3 = deepcopy(l2)
390 call assert_true(l3[1] is l3[2])
Bram Moolenaar44b4a242020-09-05 17:18:28 +0200391 call assert_fails("call deepcopy([1, 2], 2)", 'E1023:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100392endfunc
393
394" Locked variables
395func Test_list_locked_var()
396 let expected = [
Bram Moolenaara187c432020-09-16 21:08:28 +0200397 \ [['1000-000', 'ppppppF'],
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100398 \ ['0000-000', 'ppppppp'],
399 \ ['0000-000', 'ppppppp']],
400 \ [['1000-000', 'ppppppF'],
401 \ ['0000-000', 'ppppppp'],
402 \ ['0000-000', 'ppppppp']],
403 \ [['1100-100', 'ppFppFF'],
404 \ ['0000-000', 'ppppppp'],
405 \ ['0000-000', 'ppppppp']],
406 \ [['1110-110', 'pFFpFFF'],
407 \ ['0010-010', 'pFppFpp'],
408 \ ['0000-000', 'ppppppp']],
409 \ [['1111-111', 'FFFFFFF'],
410 \ ['0011-011', 'FFpFFpp'],
411 \ ['0000-000', 'ppppppp']]
412 \ ]
413 for depth in range(5)
414 for u in range(3)
415 unlet! l
416 let l = [0, [1, [2, 3]], {4: 5, 6: {7: 8}}]
417 exe "lockvar " . depth . " l"
418 if u == 1
419 exe "unlockvar l"
420 elseif u == 2
421 exe "unlockvar " . depth . " l"
422 endif
423 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]")
Bram Moolenaara187c432020-09-16 21:08:28 +0200424 call assert_equal(expected[depth][u][0], ps, 'depth: ' .. depth)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100425 let ps = ''
426 try
427 let l[1][1][0] = 99
428 let ps .= 'p'
429 catch
430 let ps .= 'F'
431 endtry
432 try
433 let l[1][1] = [99]
434 let ps .= 'p'
435 catch
436 let ps .= 'F'
437 endtry
438 try
439 let l[1] = [99]
440 let ps .= 'p'
441 catch
442 let ps .= 'F'
443 endtry
444 try
445 let l[2]['6'][7] = 99
446 let ps .= 'p'
447 catch
448 let ps .= 'F'
449 endtry
450 try
451 let l[2][6] = {99: 99}
452 let ps .= 'p'
453 catch
454 let ps .= 'F'
455 endtry
456 try
457 let l[2] = {99: 99}
458 let ps .= 'p'
459 catch
460 let ps .= 'F'
461 endtry
462 try
463 let l = [99]
464 let ps .= 'p'
465 catch
466 let ps .= 'F'
467 endtry
Bram Moolenaara187c432020-09-16 21:08:28 +0200468 call assert_equal(expected[depth][u][1], ps, 'depth: ' .. depth)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100469 endfor
470 endfor
Bram Moolenaar0e05de42020-03-25 22:23:46 +0100471 call assert_fails("let x=islocked('a b')", 'E488:')
472 let mylist = [1, 2, 3]
473 call assert_fails("let x = islocked('mylist[1:2]')", 'E786:')
474 let mydict = {'k' : 'v'}
475 call assert_fails("let x = islocked('mydict.a')", 'E716:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100476endfunc
477
478" Unletting locked variables
479func Test_list_locked_var_unlet()
480 let expected = [
Bram Moolenaara187c432020-09-16 21:08:28 +0200481 \ [['1000-000', 'ppppppp'],
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100482 \ ['0000-000', 'ppppppp'],
483 \ ['0000-000', 'ppppppp']],
484 \ [['1000-000', 'ppFppFp'],
485 \ ['0000-000', 'ppppppp'],
486 \ ['0000-000', 'ppppppp']],
487 \ [['1100-100', 'pFFpFFp'],
488 \ ['0000-000', 'ppppppp'],
489 \ ['0000-000', 'ppppppp']],
490 \ [['1110-110', 'FFFFFFp'],
491 \ ['0010-010', 'FppFppp'],
492 \ ['0000-000', 'ppppppp']],
493 \ [['1111-111', 'FFFFFFp'],
494 \ ['0011-011', 'FppFppp'],
495 \ ['0000-000', 'ppppppp']]
496 \ ]
497
498 for depth in range(5)
499 for u in range(3)
500 unlet! l
501 let l = [0, [1, [2, 3]], {4: 5, 6: {7: 8}}]
502 exe "lockvar " . depth . " l"
503 if u == 1
504 exe "unlockvar l"
505 elseif u == 2
506 exe "unlockvar " . depth . " l"
507 endif
508 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]")
Bram Moolenaara187c432020-09-16 21:08:28 +0200509 call assert_equal(expected[depth][u][0], ps, 'depth: ' .. depth)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100510 let ps = ''
511 try
512 unlet l[2]['6'][7]
513 let ps .= 'p'
514 catch
515 let ps .= 'F'
516 endtry
517 try
518 unlet l[2][6]
519 let ps .= 'p'
520 catch
521 let ps .= 'F'
522 endtry
523 try
524 unlet l[2]
525 let ps .= 'p'
526 catch
527 let ps .= 'F'
528 endtry
529 try
530 unlet l[1][1][0]
531 let ps .= 'p'
532 catch
533 let ps .= 'F'
534 endtry
535 try
536 unlet l[1][1]
537 let ps .= 'p'
538 catch
539 let ps .= 'F'
540 endtry
541 try
542 unlet l[1]
543 let ps .= 'p'
544 catch
545 let ps .= 'F'
546 endtry
547 try
548 unlet l
549 let ps .= 'p'
550 catch
551 let ps .= 'F'
552 endtry
553 call assert_equal(expected[depth][u][1], ps)
554 endfor
555 endfor
Yegappan Lakshmanan34fcb692021-05-25 20:14:00 +0200556 " Deleting a list range should fail if the range is locked
557 let l = [1, 2, 3, 4]
558 lockvar l[1:2]
559 call assert_fails('unlet l[1:2]', 'E741:')
560 unlet l
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100561endfunc
562
563" Locked variables and :unlet or list / dict functions
564
565" No :unlet after lock on dict:
566func Test_dict_lock_unlet()
567 unlet! d
568 let d = {'a': 99, 'b': 100}
569 lockvar 1 d
Bram Moolenaare2e40752020-09-04 21:18:46 +0200570 call assert_fails('unlet d.a', 'E741:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100571endfunc
572
573" unlet after lock on dict item
574func Test_dict_item_lock_unlet()
575 unlet! d
576 let d = {'a': 99, 'b': 100}
577 lockvar d.a
578 unlet d.a
579 call assert_equal({'b' : 100}, d)
580endfunc
581
582" filter() after lock on dict item
583func Test_dict_lock_filter()
584 unlet! d
585 let d = {'a': 99, 'b': 100}
586 lockvar d.a
587 call filter(d, 'v:key != "a"')
588 call assert_equal({'b' : 100}, d)
589endfunc
590
591" map() after lock on dict
592func Test_dict_lock_map()
593 unlet! d
594 let d = {'a': 99, 'b': 100}
595 lockvar 1 d
596 call map(d, 'v:val + 200')
597 call assert_equal({'a' : 299, 'b' : 300}, d)
598endfunc
599
600" No extend() after lock on dict item
601func Test_dict_lock_extend()
602 unlet! d
603 let d = {'a': 99, 'b': 100}
604 lockvar d.a
Bram Moolenaare2e40752020-09-04 21:18:46 +0200605 call assert_fails("call extend(d, {'a' : 123})", 'E741:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100606 call assert_equal({'a': 99, 'b': 100}, d)
607endfunc
608
Bram Moolenaarf7b398c2020-04-23 15:46:35 +0200609" Cannot use += with a locked dict
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200610func Test_dict_lock_operator()
611 unlet! d
612 let d = {}
613 lockvar d
614 call assert_fails("let d += {'k' : 10}", 'E741:')
615 unlockvar d
616endfunc
617
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100618" No remove() of write-protected scope-level variable
Bram Moolenaar1e115362019-01-09 23:01:02 +0100619func Tfunc1(this_is_a_long_parameter_name)
Bram Moolenaare2e40752020-09-04 21:18:46 +0200620 call assert_fails("call remove(a:, 'this_is_a_long_parameter_name')", 'E742:')
Bram Moolenaar1e115362019-01-09 23:01:02 +0100621endfunc
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100622func Test_dict_scope_var_remove()
Bram Moolenaar1e115362019-01-09 23:01:02 +0100623 call Tfunc1('testval')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100624endfunc
625
626" No extend() of write-protected scope-level variable
Bram Moolenaar31b81602019-02-10 22:14:27 +0100627func Test_dict_scope_var_extend()
Bram Moolenaare2e40752020-09-04 21:18:46 +0200628 call assert_fails("call extend(a:, {'this_is_a_long_parameter_name': 1234})", 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +0100629endfunc
630
Bram Moolenaar1e115362019-01-09 23:01:02 +0100631func Tfunc2(this_is_a_long_parameter_name)
Bram Moolenaare2e40752020-09-04 21:18:46 +0200632 call assert_fails("call extend(a:, {'this_is_a_long_parameter_name': 1234})", 'E742:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100633endfunc
Bram Moolenaar31b81602019-02-10 22:14:27 +0100634func Test_dict_scope_var_extend_overwrite()
Bram Moolenaar1e115362019-01-09 23:01:02 +0100635 call Tfunc2('testval')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100636endfunc
637
638" No :unlet of variable in locked scope
639func Test_lock_var_unlet()
640 let b:testvar = 123
641 lockvar 1 b:
642 call assert_fails('unlet b:testvar', 'E741:')
643 unlockvar 1 b:
644 unlet! b:testvar
645endfunc
646
647" No :let += of locked list variable
648func Test_let_lock_list()
649 let l = ['a', 'b', 3]
650 lockvar 1 l
651 call assert_fails("let l += ['x']", 'E741:')
652 call assert_equal(['a', 'b', 3], l)
653
654 unlet l
655 let l = [1, 2, 3, 4]
656 lockvar! l
657 call assert_equal([1, 2, 3, 4], l)
658 unlockvar l[1]
659 call assert_fails('unlet l[0:1]', 'E741:')
660 call assert_equal([1, 2, 3, 4], l)
661 call assert_fails('unlet l[1:2]', 'E741:')
662 call assert_equal([1, 2, 3, 4], l)
663 unlockvar l[1]
664 call assert_fails('let l[0:1] = [0, 1]', 'E741:')
665 call assert_equal([1, 2, 3, 4], l)
666 call assert_fails('let l[1:2] = [0, 1]', 'E741:')
667 call assert_equal([1, 2, 3, 4], l)
668 unlet l
669endfunc
670
Bram Moolenaar8dfcce32020-03-18 19:32:26 +0100671" Locking part of the list
672func Test_let_lock_list_items()
673 let l = [1, 2, 3, 4]
674 lockvar l[2:]
675 call assert_equal(0, islocked('l[0]'))
676 call assert_equal(1, islocked('l[2]'))
677 call assert_equal(1, islocked('l[3]'))
678 call assert_fails('let l[2] = 10', 'E741:')
679 call assert_fails('let l[3] = 20', 'E741:')
680 unlet l
681endfunc
682
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100683" lockvar/islocked() triggering script autoloading
684func Test_lockvar_script_autoload()
685 let old_rtp = &rtp
686 set rtp+=./sautest
687 lockvar g:footest#x
688 unlockvar g:footest#x
Bram Moolenaarf9f24ce2019-08-31 21:17:39 +0200689 call assert_equal(-1, 'g:footest#x'->islocked())
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100690 call assert_equal(0, exists('g:footest#x'))
691 call assert_equal(1, g:footest#x)
692 let &rtp = old_rtp
693endfunc
694
695" a:000 function argument test
696func s:arg_list_test(...)
697 call assert_fails('let a:000 = [1, 2]', 'E46:')
698 call assert_fails('let a:000[0] = 9', 'E742:')
699 call assert_fails('let a:000[2] = [9, 10]', 'E742:')
700 call assert_fails('let a:000[3] = {9 : 10}', 'E742:')
701
702 " now the tests that should pass
703 let a:000[2][1] = 9
704 call extend(a:000[2], [5, 6])
705 let a:000[3][5] = 8
706 let a:000[3]['a'] = 12
707 call assert_equal([1, 2, [3, 9, 5, 6], {'a': 12, '5': 8}], a:000)
708endfunc
709
710func Test_func_arg_list()
711 call s:arg_list_test(1, 2, [3, 4], {5: 6})
712endfunc
713
Bram Moolenaara187c432020-09-16 21:08:28 +0200714func Test_dict_item_locked()
715endfunc
716
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100717" Tests for reverse(), sort(), uniq()
718func Test_reverse_sort_uniq()
719 let l = ['-0', 'A11', 2, 2, 'xaaa', 4, 'foo', 'foo6', 'foo', [0, 1, 2], 'x8', [0, 1, 2], 1.5]
720 call assert_equal(['-0', 'A11', 2, 'xaaa', 4, 'foo', 'foo6', 'foo', [0, 1, 2], 'x8', [0, 1, 2], 1.5], uniq(copy(l)))
721 call assert_equal([1.5, [0, 1, 2], 'x8', [0, 1, 2], 'foo', 'foo6', 'foo', 4, 'xaaa', 2, 2, 'A11', '-0'], reverse(l))
722 call assert_equal([1.5, [0, 1, 2], 'x8', [0, 1, 2], 'foo', 'foo6', 'foo', 4, 'xaaa', 2, 2, 'A11', '-0'], reverse(reverse(l)))
Bram Moolenaar5feabe02020-01-30 18:24:53 +0100723 if has('float')
724 call assert_equal(['-0', 'A11', 'foo', 'foo', 'foo6', 'x8', 'xaaa', 1.5, 2, 2, 4, [0, 1, 2], [0, 1, 2]], sort(l))
725 call assert_equal([[0, 1, 2], [0, 1, 2], 4, 2, 2, 1.5, 'xaaa', 'x8', 'foo6', 'foo', 'foo', 'A11', '-0'], reverse(sort(l)))
726 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))))
727 call assert_equal(['-0', 'A11', 'foo', 'foo6', 'x8', 'xaaa', 1.5, 2, 4, [0, 1, 2]], uniq(sort(l)))
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100728
Bram Moolenaar5feabe02020-01-30 18:24:53 +0100729 let l = [7, 9, 'one', 18, 12, 22, 'two', 10.0e-16, -1, 'three', 0xff, 0.22, 'four']
730 call assert_equal([-1, 'one', 'two', 'three', 'four', 1.0e-15, 0.22, 7, 9, 12, 18, 22, 255], sort(copy(l), 'n'))
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100731
Bram Moolenaar5feabe02020-01-30 18:24:53 +0100732 let l = [7, 9, 18, 12, 22, 10.0e-16, -1, 0xff, 0, -0, 0.22, 'bar', 'BAR', 'Bar', 'Foo', 'FOO', 'foo', 'FOOBAR', {}, []]
733 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))
734 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'))
735 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)))
736 endif
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +0100737
Bram Moolenaarbf821bc2019-01-23 21:15:02 +0100738 call assert_fails('call reverse("")', 'E899:')
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200739 call assert_fails('call uniq([1, 2], {x, y -> []})', 'E745:')
Bram Moolenaar08f41572020-04-20 16:50:00 +0200740 call assert_fails("call sort([1, 2], function('min'), 1)", "E715:")
741 call assert_fails("call sort([1, 2], function('invalid_func'))", "E700:")
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200742 call assert_fails("call sort([1, 2], function('min'))", "E118:")
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100743endfunc
744
Bram Moolenaar85629982020-06-01 18:39:20 +0200745" reduce a list or a blob
746func Test_reduce()
747 call assert_equal(1, reduce([], { acc, val -> acc + val }, 1))
748 call assert_equal(10, reduce([1, 3, 5], { acc, val -> acc + val }, 1))
749 call assert_equal(2 * (2 * ((2 * 1) + 2) + 3) + 4, reduce([2, 3, 4], { acc, val -> 2 * acc + val }, 1))
750 call assert_equal('a x y z', ['x', 'y', 'z']->reduce({ acc, val -> acc .. ' ' .. val}, 'a'))
751 call assert_equal(#{ x: 1, y: 1, z: 1 }, ['x', 'y', 'z']->reduce({ acc, val -> extend(acc, { val: 1 }) }, {}))
752 call assert_equal([0, 1, 2, 3], reduce([1, 2, 3], function('add'), [0]))
753
754 let l = ['x', 'y', 'z']
755 call assert_equal(42, reduce(l, function('get'), #{ x: #{ y: #{ z: 42 } } }))
756 call assert_equal(['x', 'y', 'z'], l)
757
758 call assert_equal(1, reduce([1], { acc, val -> acc + val }))
759 call assert_equal('x y z', reduce(['x', 'y', 'z'], { acc, val -> acc .. ' ' .. val }))
760 call assert_equal(120, range(1, 5)->reduce({ acc, val -> acc * val }))
761 call assert_fails("call reduce([], { acc, val -> acc + val })", 'E998: Reduce of an empty List with no initial value')
762
763 call assert_equal(1, reduce(0z, { acc, val -> acc + val }, 1))
764 call assert_equal(1 + 0xaf + 0xbf + 0xcf, reduce(0zAFBFCF, { acc, val -> acc + val }, 1))
765 call assert_equal(2 * (2 * 1 + 0xaf) + 0xbf, 0zAFBF->reduce({ acc, val -> 2 * acc + val }, 1))
766
767 call assert_equal(0xff, reduce(0zff, { acc, val -> acc + val }))
768 call assert_equal(2 * (2 * 0xaf + 0xbf) + 0xcf, reduce(0zAFBFCF, { acc, val -> 2 * acc + val }))
769 call assert_fails("call reduce(0z, { acc, val -> acc + val })", 'E998: Reduce of an empty Blob with no initial value')
770
771 call assert_fails("call reduce({}, { acc, val -> acc + val }, 1)", 'E897:')
772 call assert_fails("call reduce(0, { acc, val -> acc + val }, 1)", 'E897:')
773 call assert_fails("call reduce('', { acc, val -> acc + val }, 1)", 'E897:')
Bram Moolenaarafe8cf62020-10-05 20:07:18 +0200774 call assert_fails("call reduce([1, 2], 'Xdoes_not_exist')", 'E117:')
775 call assert_fails("echo reduce(0z01, { acc, val -> 2 * acc + val }, '')", 'E39:')
Bram Moolenaarca275a02020-06-24 22:07:46 +0200776
777 let g:lut = [1, 2, 3, 4]
778 func EvilRemove()
779 call remove(g:lut, 1)
780 return 1
781 endfunc
782 call assert_fails("call reduce(g:lut, { acc, val -> EvilRemove() }, 1)", 'E742:')
783 unlet g:lut
784 delfunc EvilRemove
Bram Moolenaarfda20c42020-06-29 20:09:36 +0200785
786 call assert_equal(42, reduce(test_null_list(), function('add'), 42))
787 call assert_equal(42, reduce(test_null_blob(), function('add'), 42))
Bram Moolenaar0d90e722020-11-03 18:20:19 +0100788
789 " should not crash
790 call assert_fails('echo reduce([1], test_null_function())', 'E1132:')
Dominique Pellefe8ebdb2021-05-13 14:55:55 +0200791 call assert_fails('echo reduce([1], test_null_partial())', 'E1132:')
Bram Moolenaar85629982020-06-01 18:39:20 +0200792endfunc
793
Bram Moolenaar0e05de42020-03-25 22:23:46 +0100794" splitting a string to a List using split()
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100795func Test_str_split()
796 call assert_equal(['aa', 'bb'], split(' aa bb '))
797 call assert_equal(['aa', 'bb'], split(' aa bb ', '\W\+', 0))
798 call assert_equal(['', 'aa', 'bb', ''], split(' aa bb ', '\W\+', 1))
799 call assert_equal(['', '', 'aa', '', 'bb', '', ''], split(' aa bb ', '\W', 1))
800 call assert_equal(['aa', '', 'bb'], split(':aa::bb:', ':', 0))
801 call assert_equal(['', 'aa', '', 'bb', ''], split(':aa::bb:', ':', 1))
802 call assert_equal(['aa', '', 'bb', 'cc', ''], split('aa,,bb, cc,', ',\s*', 1))
803 call assert_equal(['a', 'b', 'c'], split('abc', '\zs'))
804 call assert_equal(['', 'a', '', 'b', '', 'c', ''], split('abc', '\zs', 1))
Bram Moolenaar0e05de42020-03-25 22:23:46 +0100805 call assert_fails("call split('abc', [])", 'E730:')
806 call assert_fails("call split('abc', 'b', [])", 'E745:')
Bram Moolenaar531be472020-09-23 22:38:05 +0200807 call assert_equal(['abc'], split('abc', '\\%('))
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100808endfunc
809
810" compare recursively linked list and dict
811func Test_listdict_compare()
812 let l = [1, 2, 3, 4]
813 let d = {'1': 1, '2': l, '3': 3}
814 let l[1] = d
815 call assert_true(l == l)
816 call assert_true(d == d)
817 call assert_false(l != deepcopy(l))
818 call assert_false(d != deepcopy(d))
Bram Moolenaar8b633132020-03-20 18:20:51 +0100819
820 " comparison errors
821 call assert_fails('echo [1, 2] =~ {}', 'E691:')
822 call assert_fails('echo [1, 2] =~ [1, 2]', 'E692:')
823 call assert_fails('echo {} =~ 5', 'E735:')
824 call assert_fails('echo {} =~ {}', 'E736:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100825endfunc
826
827 " compare complex recursively linked list and dict
828func Test_listdict_compare_complex()
829 let l = []
830 call add(l, l)
831 let dict4 = {"l": l}
832 call add(dict4.l, dict4)
833 let lcopy = deepcopy(l)
834 let dict4copy = deepcopy(dict4)
835 call assert_true(l == lcopy)
836 call assert_true(dict4 == dict4copy)
837endfunc
838
Bram Moolenaar92b83cc2020-04-25 15:24:44 +0200839" Test for extending lists and dictionaries
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100840func Test_listdict_extend()
Bram Moolenaar58d63a02019-02-25 05:56:31 +0100841 " Test extend() with lists
842
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100843 " Pass the same List to extend()
Bram Moolenaar58d63a02019-02-25 05:56:31 +0100844 let l = [1, 2, 3]
845 call assert_equal([1, 2, 3, 1, 2, 3], extend(l, l))
846 call assert_equal([1, 2, 3, 1, 2, 3], l)
847
848 let l = [1, 2, 3]
849 call assert_equal([1, 2, 3, 4, 5, 6], extend(l, [4, 5, 6]))
850 call assert_equal([1, 2, 3, 4, 5, 6], l)
851
852 let l = [1, 2, 3]
853 call extend(l, [4, 5, 6], 0)
854 call assert_equal([4, 5, 6, 1, 2, 3], l)
855
856 let l = [1, 2, 3]
857 call extend(l, [4, 5, 6], 1)
858 call assert_equal([1, 4, 5, 6, 2, 3], l)
859
860 let l = [1, 2, 3]
861 call extend(l, [4, 5, 6], 3)
862 call assert_equal([1, 2, 3, 4, 5, 6], l)
863
864 let l = [1, 2, 3]
865 call extend(l, [4, 5, 6], -1)
866 call assert_equal([1, 2, 4, 5, 6, 3], l)
867
868 let l = [1, 2, 3]
869 call extend(l, [4, 5, 6], -3)
870 call assert_equal([4, 5, 6, 1, 2, 3], l)
871
872 let l = [1, 2, 3]
873 call assert_fails("call extend(l, [4, 5, 6], 4)", 'E684:')
874 call assert_fails("call extend(l, [4, 5, 6], -4)", 'E684:')
Bram Moolenaar5feabe02020-01-30 18:24:53 +0100875 if has('float')
876 call assert_fails("call extend(l, [4, 5, 6], 1.2)", 'E805:')
877 endif
Bram Moolenaar58d63a02019-02-25 05:56:31 +0100878
879 " Test extend() with dictionaries.
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100880
881 " Pass the same Dict to extend()
882 let d = { 'a': {'b': 'B'}}
883 call extend(d, d)
884 call assert_equal({'a': {'b': 'B'}}, d)
885
Bram Moolenaar58d63a02019-02-25 05:56:31 +0100886 let d = {'a': 'A', 'b': 'B'}
887 call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, extend(d, {'b': 0, 'c':'C'}))
888 call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, d)
889
890 let d = {'a': 'A', 'b': 'B'}
891 call extend(d, {'a': 'A', 'b': 0, 'c': 'C'}, "force")
892 call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, d)
893
894 let d = {'a': 'A', 'b': 'B'}
895 call extend(d, {'b': 0, 'c':'C'}, "keep")
896 call assert_equal({'a': 'A', 'b': 'B', 'c': 'C'}, d)
897
898 let d = {'a': 'A', 'b': 'B'}
899 call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 'error')", 'E737:')
900 call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 'xxx')", 'E475:')
Bram Moolenaar5feabe02020-01-30 18:24:53 +0100901 if has('float')
Bram Moolenaar3cfa5b12021-06-06 14:14:39 +0200902 call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 1.2)", 'E475:')
Bram Moolenaar5feabe02020-01-30 18:24:53 +0100903 endif
Bram Moolenaar58d63a02019-02-25 05:56:31 +0100904 call assert_equal({'a': 'A', 'b': 'B'}, d)
905
906 call assert_fails("call extend([1, 2], 1)", 'E712:')
907 call assert_fails("call extend([1, 2], {})", 'E712:')
Bram Moolenaar08f41572020-04-20 16:50:00 +0200908
909 " Extend g: dictionary with an invalid variable name
910 call assert_fails("call extend(g:, {'-!' : 10})", 'E461:')
Bram Moolenaardcae51f2021-04-08 20:10:10 +0200911
912 " Extend a list with itself.
913 let l = [1, 5, 7]
914 call extend(l, l, 0)
915 call assert_equal([1, 5, 7, 1, 5, 7], l)
916 let l = [1, 5, 7]
917 call extend(l, l, 1)
918 call assert_equal([1, 1, 5, 7, 5, 7], l)
919 let l = [1, 5, 7]
920 call extend(l, l, 2)
921 call assert_equal([1, 5, 1, 5, 7, 7], l)
922 let l = [1, 5, 7]
923 call extend(l, l, 3)
924 call assert_equal([1, 5, 7, 1, 5, 7], l)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100925endfunc
Bram Moolenaar31b81602019-02-10 22:14:27 +0100926
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100927func Test_listdict_extendnew()
928 " Test extendnew() with lists
929 let l = [1, 2, 3]
930 call assert_equal([1, 2, 3, 4, 5], extendnew(l, [4, 5]))
931 call assert_equal([1, 2, 3], l)
932
933 " Test extend() with dictionaries.
934 let d = {'a': {'b': 'B'}}
935 call assert_equal({'a': {'b': 'B'}, 'c': 'cc'}, extendnew(d, {'c': 'cc'}))
936 call assert_equal({'a': {'b': 'B'}}, d)
937endfunc
938
Bram Moolenaar31b81602019-02-10 22:14:27 +0100939func s:check_scope_dict(x, fixed)
940 func s:gen_cmd(cmd, x)
941 return substitute(a:cmd, '\<x\ze:', a:x, 'g')
942 endfunc
943
944 let cmd = s:gen_cmd('let x:foo = 1', a:x)
945 if a:fixed
Bram Moolenaare2e40752020-09-04 21:18:46 +0200946 call assert_fails(cmd, 'E461:')
Bram Moolenaar31b81602019-02-10 22:14:27 +0100947 else
948 exe cmd
949 exe s:gen_cmd('call assert_equal(1, x:foo)', a:x)
950 endif
951
952 let cmd = s:gen_cmd('let x:["bar"] = 2', a:x)
953 if a:fixed
Bram Moolenaare2e40752020-09-04 21:18:46 +0200954 call assert_fails(cmd, 'E461:')
Bram Moolenaar31b81602019-02-10 22:14:27 +0100955 else
956 exe cmd
957 exe s:gen_cmd('call assert_equal(2, x:bar)', a:x)
958 endif
959
960 let cmd = s:gen_cmd('call extend(x:, {"baz": 3})', a:x)
961 if a:fixed
Bram Moolenaare2e40752020-09-04 21:18:46 +0200962 call assert_fails(cmd, 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +0100963 else
964 exe cmd
965 exe s:gen_cmd('call assert_equal(3, x:baz)', a:x)
966 endif
967
968 if a:fixed
969 if a:x ==# 'a'
Bram Moolenaare2e40752020-09-04 21:18:46 +0200970 call assert_fails('unlet a:x', 'E795:')
971 call assert_fails('call remove(a:, "x")', 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +0100972 elseif a:x ==# 'v'
Bram Moolenaare2e40752020-09-04 21:18:46 +0200973 call assert_fails('unlet v:count', 'E795:')
974 call assert_fails('call remove(v:, "count")', 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +0100975 endif
976 else
977 exe s:gen_cmd('unlet x:foo', a:x)
978 exe s:gen_cmd('unlet x:bar', a:x)
979 exe s:gen_cmd('call remove(x:, "baz")', a:x)
980 endif
981
982 delfunc s:gen_cmd
983endfunc
984
985func Test_scope_dict()
986 " Test for g:
987 call s:check_scope_dict('g', v:false)
988
989 " Test for s:
990 call s:check_scope_dict('s', v:false)
991
992 " Test for l:
993 call s:check_scope_dict('l', v:false)
994
995 " Test for a:
996 call s:check_scope_dict('a', v:true)
997
998 " Test for b:
999 call s:check_scope_dict('b', v:false)
1000
1001 " Test for w:
1002 call s:check_scope_dict('w', v:false)
1003
1004 " Test for t:
1005 call s:check_scope_dict('t', v:false)
1006
1007 " Test for v:
1008 call s:check_scope_dict('v', v:true)
1009endfunc
Bram Moolenaar8dfcce32020-03-18 19:32:26 +01001010
1011" Test for deep nesting of lists (> 100)
1012func Test_deep_nested_list()
1013 let deep_list = []
1014 let l = deep_list
1015 for i in range(102)
1016 let newlist = []
1017 call add(l, newlist)
1018 let l = newlist
1019 endfor
1020 call add(l, 102)
1021
1022 call assert_fails('let m = deepcopy(deep_list)', 'E698:')
1023 call assert_fails('lockvar 110 deep_list', 'E743:')
1024 call assert_fails('unlockvar 110 deep_list', 'E743:')
1025 call assert_fails('let x = execute("echo deep_list")', 'E724:')
1026 call test_garbagecollect_now()
1027 unlet deep_list
1028endfunc
1029
1030" Test for deep nesting of dicts (> 100)
1031func Test_deep_nested_dict()
1032 let deep_dict = {}
1033 let d = deep_dict
1034 for i in range(102)
1035 let newdict = {}
1036 let d.k = newdict
1037 let d = newdict
1038 endfor
1039 let d.k = 'v'
1040
1041 call assert_fails('let m = deepcopy(deep_dict)', 'E698:')
1042 call assert_fails('lockvar 110 deep_dict', 'E743:')
1043 call assert_fails('unlockvar 110 deep_dict', 'E743:')
1044 call assert_fails('let x = execute("echo deep_dict")', 'E724:')
1045 call test_garbagecollect_now()
1046 unlet deep_dict
1047endfunc
1048
Bram Moolenaar8b633132020-03-20 18:20:51 +01001049" List and dict indexing tests
1050func Test_listdict_index()
1051 call assert_fails('echo function("min")[0]', 'E695:')
1052 call assert_fails('echo v:true[0]', 'E909:')
1053 let d = {'k' : 10}
1054 call assert_fails('echo d.', 'E15:')
1055 call assert_fails('echo d[1:2]', 'E719:')
1056 call assert_fails("let v = [4, 6][{-> 1}]", 'E729:')
1057 call assert_fails("let v = range(5)[2:[]]", 'E730:')
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001058 call assert_fails("let v = range(5)[2:{-> 2}(]", ['E15:', 'E116:'])
Bram Moolenaar8b633132020-03-20 18:20:51 +01001059 call assert_fails("let v = range(5)[2:3", 'E111:')
Bram Moolenaar08f41572020-04-20 16:50:00 +02001060 call assert_fails("let l = insert([1,2,3], 4, 10)", 'E684:')
1061 call assert_fails("let l = insert([1,2,3], 4, -10)", 'E684:')
1062 call assert_fails("let l = insert([1,2,3], 4, [])", 'E745:')
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001063 let l = [1, 2, 3]
1064 call assert_fails("let l[i] = 3", 'E121:')
Bram Moolenaar3cfa5b12021-06-06 14:14:39 +02001065 call assert_fails("let l[1.1] = 4", 'E805:')
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001066 call assert_fails("let l[:i] = [4, 5]", 'E121:')
Bram Moolenaar3cfa5b12021-06-06 14:14:39 +02001067 call assert_fails("let l[:3.2] = [4, 5]", 'E805:')
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001068 let t = test_unknown()
1069 call assert_fails("echo t[0]", 'E685:')
Bram Moolenaar08f41572020-04-20 16:50:00 +02001070endfunc
1071
1072" Test for a null list
1073func Test_null_list()
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001074 let l = test_null_list()
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001075 call assert_equal(0, join(test_null_list()))
1076 call assert_equal('', join(l))
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001077 call assert_equal(0, len(l))
1078 call assert_equal(1, empty(l))
Bram Moolenaar08f41572020-04-20 16:50:00 +02001079 call assert_fails('let s = join([1, 2], [])', 'E730:')
1080 call assert_equal([], split(test_null_string()))
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001081 call assert_equal([], l[:2])
1082 call assert_true([] == l)
1083 call assert_equal('[]', string(l))
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001084 call assert_equal(0, sort(test_null_list()))
1085 call assert_equal([], sort(l))
1086 call assert_equal(0, uniq(test_null_list()))
1087 call assert_equal([], uniq(l))
Bram Moolenaarf5749722020-10-10 16:42:39 +02001088 let k = [] + l
1089 call assert_equal([], k)
1090 let k = l + []
1091 call assert_equal([], k)
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001092 call assert_equal(0, len(copy(l)))
1093 call assert_equal(0, count(l, 5))
1094 call assert_equal([], deepcopy(l))
1095 call assert_equal(5, get(l, 2, 5))
1096 call assert_equal(-1, index(l, 2, 5))
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001097 call assert_equal(0, insert(test_null_list(), 2, -1))
1098 call assert_fails('call insert(l, 2, -1)', 'E684:')
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001099 call assert_equal(0, min(l))
1100 call assert_equal(0, max(l))
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001101 call assert_equal(0, remove(test_null_list(), 0, 2))
1102 call assert_fails('call remove(l, 0, 2)', 'E684:')
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001103 call assert_equal([], repeat(l, 2))
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001104 call assert_equal(0, reverse(test_null_list()))
1105 call assert_equal([], reverse(l))
1106 call assert_equal(0, sort(test_null_list()))
1107 call assert_equal([], sort(l))
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001108 call assert_equal('[]', string(l))
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001109 call assert_fails('call extend(test_null_list(), test_null_list())', 'E1134:')
1110 call assert_equal([], extend(l, l, 0))
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001111 lockvar l
1112 call assert_equal(1, islocked('l'))
1113 unlockvar l
Bram Moolenaar08f41572020-04-20 16:50:00 +02001114endfunc
1115
1116" Test for a null dict
1117func Test_null_dict()
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001118 call assert_equal(test_null_dict(), test_null_dict())
1119 let d = test_null_dict()
1120 call assert_equal({}, d)
1121 call assert_equal(0, len(d))
1122 call assert_equal(1, empty(d))
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001123 call assert_equal(0, items(test_null_dict()))
1124 call assert_equal([], items(d))
1125 call assert_equal(0, keys(test_null_dict()))
1126 call assert_equal([], keys(d))
1127 call assert_equal(0, values(test_null_dict()))
1128 call assert_equal([], values(d))
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001129 call assert_false(has_key(d, 'k'))
1130 call assert_equal('{}', string(d))
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001131 call assert_fails('let x = d[10]', 'E716:')
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001132 call assert_equal({}, {})
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001133 call assert_equal(0, len(copy(d)))
1134 call assert_equal(0, count(d, 'k'))
1135 call assert_equal({}, deepcopy(d))
1136 call assert_equal(20, get(d, 'k', 20))
1137 call assert_equal(0, min(d))
1138 call assert_equal(0, max(d))
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001139 call assert_equal(0, remove(test_null_dict(), 'k'))
1140 call assert_fails("call remove(d, 'k')", 'E716:')
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001141 call assert_equal('{}', string(d))
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001142 call assert_fails('call extend(test_null_dict(), test_null_dict())', 'E1133:')
1143 call assert_equal({}, extend(d, d, 'keep'))
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001144 lockvar d
1145 call assert_equal(1, islocked('d'))
1146 unlockvar d
Bram Moolenaar8b633132020-03-20 18:20:51 +01001147endfunc
1148
Bram Moolenaar8dfcce32020-03-18 19:32:26 +01001149" vim: shiftwidth=2 sts=2 expandtab