blob: 957520bcc80a2f3dbdff7eb9a3a1594edc84c954 [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 Moolenaar92f05f22021-08-12 21:12:56 +020045
46 let lines =<< trim END
47 VAR l = [1, 2]
48 call assert_equal([1, 2], l[:])
49 call assert_equal([2], l[-1 : -1])
50 call assert_equal([1, 2], l[-2 : -1])
51 END
52 call CheckLegacyAndVim9Success(lines)
53
54 let l = [1, 2]
55 call assert_equal([], l[-3 : -1])
56
57 let lines =<< trim END
58 var l = [1, 2]
59 assert_equal([1, 2], l[-3 : -1])
60 END
61 call CheckDefAndScriptSuccess(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +010062endfunc
63
64" List identity
65func Test_list_identity()
Bram Moolenaar63cb6562021-07-20 22:21:59 +020066 let lines =<< trim END
67 VAR l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
68 VAR ll = l
69 VAR lx = copy(l)
70 call assert_true(l == ll)
71 call assert_false(l isnot ll)
72 call assert_true(l is ll)
73 call assert_true(l == lx)
74 call assert_false(l is lx)
75 call assert_true(l isnot lx)
76 END
77 call CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +010078endfunc
79
80" removing items with :unlet
81func Test_list_unlet()
Bram Moolenaar63cb6562021-07-20 22:21:59 +020082 let lines =<< trim END
83 VAR l = [1, 'as''d', [1, 2, function("strlen")], {'a': 1},]
84 unlet l[2]
85 call assert_equal([1, 'as''d', {'a': 1}], l)
86 LET l = range(8)
87 unlet l[: 3]
88 unlet l[1 :]
89 call assert_equal([4], l)
Bram Moolenaarfb094e12017-11-05 20:59:28 +010090
Bram Moolenaar63cb6562021-07-20 22:21:59 +020091 #" removing items out of range: silently skip items that don't exist
92 LET l = [0, 1, 2, 3]
93 unlet l[2 : 2]
94 call assert_equal([0, 1, 3], l)
95 LET l = [0, 1, 2, 3]
96 unlet l[2 : 3]
97 call assert_equal([0, 1], l)
98 LET l = [0, 1, 2, 3]
99 unlet l[2 : 4]
100 call assert_equal([0, 1], l)
101 LET l = [0, 1, 2, 3]
102 unlet l[2 : 5]
103 call assert_equal([0, 1], l)
104 LET l = [0, 1, 2, 3]
105 unlet l[-2 : 2]
106 call assert_equal([0, 1, 3], l)
107 LET l = [0, 1, 2, 3]
108 unlet l[-3 : 2]
109 call assert_equal([0, 3], l)
110 LET l = [0, 1, 2, 3]
111 unlet l[-4 : 2]
112 call assert_equal([3], l)
113 LET l = [0, 1, 2, 3]
114 unlet l[-5 : 2]
115 call assert_equal([3], l)
116 LET l = [0, 1, 2, 3]
117 unlet l[-6 : 2]
118 call assert_equal([3], l)
119 END
120 call CheckLegacyAndVim9Success(lines)
121
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100122 let l = [0, 1, 2, 3]
123 unlet l[2:2]
124 call assert_equal([0, 1, 3], l)
125 let l = [0, 1, 2, 3]
126 unlet l[2:3]
127 call assert_equal([0, 1], l)
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200128
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200129 let lines =<< trim END
130 VAR l = [0, 1, 2, 3]
131 unlet l[2 : 1]
132 END
133 call CheckLegacyAndVim9Failure(lines, 'E684:')
134
135 let lines =<< trim END
136 VAR l = [0, 1, 2, 3]
137 unlet l[-1 : 2]
138 END
139 call CheckLegacyAndVim9Failure(lines, 'E684:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100140endfunc
141
142" assignment to a list
143func Test_list_assign()
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200144 let lines =<< trim END
145 VAR l = [0, 1, 2, 3]
146 VAR va = 0
147 VAR vb = 0
148 LET [va, vb] = l[2 : 3]
149 call assert_equal([2, 3], [va, vb])
150 END
151 call CheckLegacyAndVim9Success(lines)
152
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200153 let lines =<< trim END
154 let l = [0, 1, 2, 3]
155 let [va, vb] = l
156 END
157 call CheckScriptFailure(lines, 'E687:')
158 let lines =<< trim END
159 var l = [0, 1, 2, 3]
160 var va = 0
161 var vb = 0
162 [va, vb] = l
163 END
164 call CheckScriptFailure(['vim9script'] + lines, 'E687:')
165 call CheckDefExecFailure(lines, 'E1093: Expected 2 items but got 4')
166
167 let lines =<< trim END
168 let l = [0, 1, 2, 3]
169 let [va, vb] = l[1:1]
170 END
171 call CheckScriptFailure(lines, 'E688:')
172 let lines =<< trim END
173 var l = [0, 1, 2, 3]
174 var va = 0
175 var vb = 0
176 [va, vb] = l[1 : 1]
177 END
178 call CheckScriptFailure(['vim9script'] + lines, 'E688:')
179 call CheckDefExecFailure(lines, 'E1093: Expected 2 items but got 1')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100180endfunc
181
182" test for range assign
183func Test_list_range_assign()
Bram Moolenaar4f0884d2021-08-11 21:49:23 +0200184 let lines =<< trim END
185 VAR l = [0]
186 LET l[:] = [1, 2]
187 call assert_equal([1, 2], l)
188 LET l[-4 : -1] = [5, 6]
189 call assert_equal([5, 6], l)
190 END
191 call CheckLegacyAndVim9Success(lines)
Bram Moolenaar89071cb2021-08-13 18:20:09 +0200192
193 let lines =<< trim END
194 var l = [7]
195 l[:] = ['text']
196 END
197 call CheckDefAndScriptFailure(lines, 'E1012:', 2)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100198endfunc
199
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200200" Test removing items in list
201func Test_list_func_remove()
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200202 let lines =<< trim END
203 #" Test removing 1 element
204 VAR l = [1, 2, 3, 4]
205 call assert_equal(1, remove(l, 0))
206 call assert_equal([2, 3, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200207
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200208 LET l = [1, 2, 3, 4]
209 call assert_equal(2, remove(l, 1))
210 call assert_equal([1, 3, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200211
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200212 LET l = [1, 2, 3, 4]
213 call assert_equal(4, remove(l, -1))
214 call assert_equal([1, 2, 3], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200215
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200216 #" Test removing range of element(s)
217 LET l = [1, 2, 3, 4]
218 call assert_equal([3], remove(l, 2, 2))
219 call assert_equal([1, 2, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200220
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200221 LET l = [1, 2, 3, 4]
222 call assert_equal([2, 3], remove(l, 1, 2))
223 call assert_equal([1, 4], l)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200224
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200225 LET l = [1, 2, 3, 4]
226 call assert_equal([2, 3], remove(l, -3, -2))
227 call assert_equal([1, 4], l)
228 END
229 call CheckLegacyAndVim9Success(lines)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200230
231 " Test invalid cases
232 let l = [1, 2, 3, 4]
233 call assert_fails("call remove(l, 5)", 'E684:')
234 call assert_fails("call remove(l, 1, 5)", 'E684:')
235 call assert_fails("call remove(l, 3, 2)", 'E16:')
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +0100236 call assert_fails("call remove(1, 0)", 'E896:')
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200237 call assert_fails("call remove(l, l)", 'E745:')
238endfunc
239
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200240" List add() function
241func Test_list_add()
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200242 let lines =<< trim END
243 VAR l = []
244 call add(l, 1)
245 call add(l, [2, 3])
246 call add(l, [])
247 call add(l, test_null_list())
248 call add(l, {'k': 3})
249 call add(l, {})
250 call add(l, test_null_dict())
251 call assert_equal([1, [2, 3], [], [], {'k': 3}, {}, {}], l)
252 END
253 call CheckLegacyAndVim9Success(lines)
254
255 " weird legacy behavior
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200256 call assert_equal(1, add(test_null_list(), 4))
257endfunc
258
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100259" Tests for Dictionary type
260
261func Test_dict()
262 " Creating Dictionary directly with different types
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200263 let lines =<< trim END
264 VAR d = {'1': 'asd', 'b': [1, 2, function('strlen')], '-1': {'a': 1}, }
265 call assert_equal("{'1': 'asd', 'b': [1, 2, function('strlen')], '-1': {'a': 1}}", string(d))
266 call assert_equal('asd', d.1)
267 call assert_equal(['-1', '1', 'b'], sort(keys(d)))
268 call assert_equal(['asd', [1, 2, function('strlen')], {'a': 1}], values(d))
269 call extend(d, {3: 33, 1: 99})
270 call extend(d, {'b': 'bbb', 'c': 'ccc'}, "keep")
271 call assert_equal({'c': 'ccc', '1': 99, 'b': [1, 2, function('strlen')], '3': 33, '-1': {'a': 1}}, d)
272 END
273 call CheckLegacyAndVim9Success(lines)
274
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100275 let d = {001: 'asd', 'b': [1, 2, function('strlen')], -1: {'a': 1},}
276 call assert_equal("{'1': 'asd', 'b': [1, 2, function('strlen')], '-1': {'a': 1}}", string(d))
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200277
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100278 let v = []
279 for [key, val] in items(d)
280 call extend(v, [key, val])
281 unlet key val
282 endfor
283 call assert_equal(['1','asd','b',[1, 2, function('strlen')],'-1',{'a': 1}], v)
284
Bram Moolenaar63cb6562021-07-20 22:21:59 +0200285 call extend(d, {3: 33, 1: 99})
Bram Moolenaare2e40752020-09-04 21:18:46 +0200286 call assert_fails("call extend(d, {3:333,4:444}, 'error')", 'E737:')
Bram Moolenaar08f41572020-04-20 16:50:00 +0200287
288 " duplicate key
289 call assert_fails("let d = {'k' : 10, 'k' : 20}", 'E721:')
290 " missing comma
291 call assert_fails("let d = {'k' : 10 'k' : 20}", 'E722:')
292 " missing curly brace
293 call assert_fails("let d = {'k' : 10,", 'E723:')
294 " invalid key
295 call assert_fails('let d = #{++ : 10}', 'E15:')
296 " wrong type for key
297 call assert_fails('let d={[] : 10}', 'E730:')
298 " undefined variable as value
299 call assert_fails("let d={'k' : i}", 'E121:')
Bram Moolenaar98cb90e2021-11-30 11:56:22 +0000300
301 " allow key starting with number at the start, not a curly expression
302 call assert_equal({'1foo': 77}, #{1foo: 77})
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100303endfunc
304
Bram Moolenaar4d4d1cd2020-07-30 22:14:33 +0200305" This was allowed in legacy Vim script
306let s:dict_with_spaces = {'one' : 1 , 'two' : 2 , 'three' : 3}
307let s:dict_with_spaces_lit = #{one : 1 , two : 2 , three : 3}
308
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100309" Dictionary identity
310func Test_dict_identity()
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200311 let lines =<< trim END
312 VAR d = {'1': 'asd', 'b': [1, 2, function('strlen')], -1: {'a': 1}, }
313 VAR dd = d
314 VAR dx = copy(d)
315 call assert_true(d == dd)
316 call assert_false(d isnot dd)
317 call assert_true(d is dd)
318 call assert_true(d == dx)
319 call assert_false(d is dx)
320 call assert_true(d isnot dx)
321 END
322 call CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100323endfunc
324
325" removing items with :unlet
326func Test_dict_unlet()
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200327 let lines =<< trim END
328 VAR d = {'b': 'bbb', '1': 99, '3': 33, '-1': {'a': 1}}
329 unlet d.b
330 unlet d[-1]
331 call assert_equal({'1': 99, '3': 33}, d)
332 END
333 call CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100334endfunc
335
336" manipulating a big Dictionary (hashtable.c has a border of 1000 entries)
337func Test_dict_big()
338 let d = {}
339 for i in range(1500)
340 let d[i] = 3000 - i
341 endfor
342 call assert_equal([3000, 2900, 2001, 1600, 1501], [d[0], d[100], d[999], d[1400], d[1499]])
343 let str = ''
344 try
345 let n = d[1500]
346 catch
Bram Moolenaar6d967122020-10-30 19:06:18 +0100347 let str = substitute(v:exception, '\v(.{14}).*( "\d{4}").*', '\1\2', '')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100348 endtry
Bram Moolenaar6d967122020-10-30 19:06:18 +0100349 call assert_equal('Vim(let):E716: "1500"', str)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100350
351 " lookup each items
352 for i in range(1500)
353 call assert_equal(3000 - i, d[i])
354 endfor
355 let i += 1
356
357 " delete even items
358 while i >= 2
359 let i -= 2
360 unlet d[i]
361 endwhile
362 call assert_equal('NONE', get(d, 1500 - 100, 'NONE'))
363 call assert_equal(2999, d[1])
364
365 " delete odd items, checking value, one intentionally wrong
366 let d[33] = 999
367 let i = 1
368 while i < 1500
369 if i != 33
370 call assert_equal(3000 - i, d[i])
371 else
372 call assert_equal(999, d[i])
373 endif
374 unlet d[i]
375 let i += 2
376 endwhile
377 call assert_equal({}, d)
378 unlet d
379endfunc
380
381" Dictionary function
382func Test_dict_func()
383 let d = {}
384 func d.func(a) dict
385 return a:a . len(self.data)
386 endfunc
387 let d.data = [1,2,3]
388 call assert_equal('len: 3', d.func('len: '))
389 let x = d.func('again: ')
390 call assert_equal('again: 3', x)
391 let Fn = d.func
392 call assert_equal('xxx3', Fn('xxx'))
393endfunc
394
Bram Moolenaarb13ab992020-07-27 21:43:28 +0200395func Test_dict_assign()
396 let d = {}
397 let d.1 = 1
398 let d._ = 2
399 call assert_equal({'1': 1, '_': 2}, d)
Bram Moolenaar3a3b10e2021-06-26 15:00:59 +0200400
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200401 let lines =<< trim END
402 VAR d = {}
403 LET d.a = 1
404 LET d._ = 2
405 call assert_equal({'a': 1, '_': 2}, d)
406 END
407 call CheckLegacyAndVim9Success(lines)
408
409 let lines =<< trim END
410 let n = 0
411 let n.key = 3
412 END
413 call CheckScriptFailure(lines, 'E1203: Dot can only be used on a dictionary: n.key = 3')
414 let lines =<< trim END
415 vim9script
416 var n = 0
417 n.key = 3
418 END
419 call CheckScriptFailure(lines, 'E1203: Dot can only be used on a dictionary: n.key = 3')
420 let lines =<< trim END
421 var n = 0
422 n.key = 3
423 END
424 call CheckDefFailure(lines, 'E1141:')
Bram Moolenaarb13ab992020-07-27 21:43:28 +0200425endfunc
426
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100427" Function in script-local List or Dict
428func Test_script_local_dict_func()
429 let g:dict = {}
430 function g:dict.func() dict
431 return 'g:dict.func' . self.foo[1] . self.foo[0]('asdf')
432 endfunc
433 let g:dict.foo = ['-', 2, 3]
434 call insert(g:dict.foo, function('strlen'))
435 call assert_equal('g:dict.func-4', g:dict.func())
436 unlet g:dict
437endfunc
438
Bram Moolenaar08f41572020-04-20 16:50:00 +0200439" Test removing items in a dictionary
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200440func Test_dict_func_remove()
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200441 let lines =<< trim END
442 VAR d = {1: 'a', 2: 'b', 3: 'c'}
443 call assert_equal('b', remove(d, 2))
444 call assert_equal({1: 'a', 3: 'c'}, d)
445 END
446 call CheckLegacyAndVim9Success(lines)
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200447
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200448 let lines =<< trim END
449 VAR d = {1: 'a', 3: 'c'}
450 call remove(d, 1, 2)
451 END
452 call CheckLegacyAndVim9Failure(lines, 'E118:')
453
454 let lines =<< trim END
455 VAR d = {1: 'a', 3: 'c'}
456 call remove(d, 'a')
457 END
458 call CheckLegacyAndVim9Failure(lines, 'E716:')
459
460 let lines =<< trim END
Bram Moolenaar5c1ec432021-11-29 13:44:55 +0000461 let d = {'a-b': 55}
462 echo d.a-b
463 END
464 call CheckScriptFailure(lines, 'E716: Key not present in Dictionary: "a"')
465
466 let lines =<< trim END
467 vim9script
468 var d = {'a-b': 55}
469 echo d.a-b
470 END
471 call CheckScriptFailure(lines, 'E716: Key not present in Dictionary: "a"')
472
473 let lines =<< trim END
474 var d = {'a-b': 55}
475 echo d.a-b
476 END
477 call CheckDefFailure(lines, 'E1004: White space required before and after ''-''')
478
479 let lines =<< trim END
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200480 let d = {1: 'a', 3: 'c'}
481 call remove(d, [])
482 END
483 call CheckScriptFailure(lines, 'E730:')
484 let lines =<< trim END
485 vim9script
486 var d = {1: 'a', 3: 'c'}
487 call remove(d, [])
488 END
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200489 call CheckScriptFailure(lines, 'E1220: String or Number required for argument 2')
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200490 let lines =<< trim END
491 var d = {1: 'a', 3: 'c'}
492 call remove(d, [])
493 END
494 call CheckDefExecFailure(lines, 'E1013: Argument 2: type mismatch, expected string but got list<unknown>')
Bram Moolenaar2bfddfc2018-09-30 17:16:25 +0200495endfunc
496
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100497" Nasty: remove func from Dict that's being called (works)
498func Test_dict_func_remove_in_use()
499 let d = {1:1}
500 func d.func(a)
501 return "a:" . a:a
502 endfunc
503 let expected = 'a:' . string(get(d, 'func'))
504 call assert_equal(expected, d.func(string(remove(d, 'func'))))
Bram Moolenaar3e9c0b92021-08-12 10:39:10 +0200505
506 " similar, in a way it also works in Vim9
507 let lines =<< trim END
508 VAR d = {1: 1, 2: 'x'}
509 func GetArg(a)
510 return "a:" .. a:a
511 endfunc
512 LET d.func = function('GetArg')
513 VAR expected = 'a:' .. string(get(d, 'func'))
514 call assert_equal(expected, d.func(string(remove(d, 'func'))))
515 END
516 call CheckTransLegacySuccess(lines)
517 call CheckTransVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100518endfunc
519
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200520func Test_dict_literal_keys()
Bram Moolenaar4c6d9042019-07-16 22:04:02 +0200521 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 +0200522
Bram Moolenaar5dd839c2021-07-22 18:48:53 +0200523 " why *{} cannot be used for a literal dictionary
Bram Moolenaarb8be54d2019-07-14 18:22:59 +0200524 let blue = 'blue'
525 call assert_equal('6', trim(execute('echo 2 *{blue: 3}.blue')))
Bram Moolenaard5abb4c2019-07-13 22:46:10 +0200526endfunc
527
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100528" Nasty: deepcopy() dict that refers to itself (fails when noref used)
529func Test_dict_deepcopy()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200530 let lines =<< trim END
531 VAR d = {1: 1, 2: '2'}
532 VAR l = [4, d, 6]
533 LET d[3] = l
534 VAR dc = deepcopy(d)
535 call deepcopy(d, 1)
536 END
537 call CheckLegacyAndVim9Failure(lines, 'E698:')
538
539 let lines =<< trim END
540 VAR d = {1: 1, 2: '2'}
541 VAR l = [4, d, 6]
542 LET d[3] = l
543 VAR l2 = [0, l, l, 3]
544 LET l[1] = l2
545 VAR l3 = deepcopy(l2)
546 call assert_true(l3[1] is l3[2])
547 END
548 call CheckLegacyAndVim9Success(lines)
549
Bram Moolenaar44b4a242020-09-05 17:18:28 +0200550 call assert_fails("call deepcopy([1, 2], 2)", 'E1023:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100551endfunc
552
553" Locked variables
554func Test_list_locked_var()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200555 " Not tested with :def function, local vars cannot be locked.
556 let lines =<< trim END
557 VAR expected = [
558 \ [['1000-000', 'ppppppF'],
559 \ ['0000-000', 'ppppppp'],
560 \ ['0000-000', 'ppppppp']],
561 \ [['1000-000', 'ppppppF'],
562 \ ['0000-000', 'ppppppp'],
563 \ ['0000-000', 'ppppppp']],
564 \ [['1100-100', 'ppFppFF'],
565 \ ['0000-000', 'ppppppp'],
566 \ ['0000-000', 'ppppppp']],
567 \ [['1110-110', 'pFFpFFF'],
568 \ ['0010-010', 'pFppFpp'],
569 \ ['0000-000', 'ppppppp']],
570 \ [['1111-111', 'FFFFFFF'],
571 \ ['0011-011', 'FFpFFpp'],
572 \ ['0000-000', 'ppppppp']]
573 \ ]
574 for depth in range(5)
575 for u in range(3)
576 VAR l = [0, [1, [2, 3]], {4: 5, 6: {7: 8}}]
577 exe "lockvar " .. depth .. " l"
578 if u == 1
579 exe "unlockvar l"
580 elseif u == 2
581 exe "unlockvar " .. depth .. " l"
582 endif
583 VAR 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]")
584 call assert_equal(expected[depth][u][0], ps, 'depth: ' .. depth)
585 LET ps = ''
586 try
587 LET l[1][1][0] = 99
588 LET ps ..= 'p'
589 catch
590 LET ps ..= 'F'
591 endtry
592 try
593 LET l[1][1] = [99]
594 LET ps ..= 'p'
595 catch
596 LET ps ..= 'F'
597 endtry
598 try
599 LET l[1] = [99]
600 LET ps ..= 'p'
601 catch
602 LET ps ..= 'F'
603 endtry
604 try
605 LET l[2]['6'][7] = 99
606 LET ps ..= 'p'
607 catch
608 LET ps ..= 'F'
609 endtry
610 try
611 LET l[2][6] = {99: 99}
612 LET ps ..= 'p'
613 catch
614 LET ps ..= 'F'
615 endtry
616 try
617 LET l[2] = {99: 99}
618 LET ps ..= 'p'
619 catch
620 LET ps ..= 'F'
621 endtry
622 try
623 LET l = [99]
624 LET ps ..= 'p'
625 catch
626 LET ps ..= 'F'
627 endtry
628 call assert_equal(expected[depth][u][1], ps, 'depth: ' .. depth)
629 unlock! l
630 endfor
631 endfor
632 END
633 call CheckTransLegacySuccess(lines)
634 call CheckTransVim9Success(lines)
635
Bram Moolenaar0e05de42020-03-25 22:23:46 +0100636 call assert_fails("let x=islocked('a b')", 'E488:')
637 let mylist = [1, 2, 3]
638 call assert_fails("let x = islocked('mylist[1:2]')", 'E786:')
639 let mydict = {'k' : 'v'}
640 call assert_fails("let x = islocked('mydict.a')", 'E716:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100641endfunc
642
643" Unletting locked variables
644func Test_list_locked_var_unlet()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200645 " Not tested with Vim9: script and local variables cannot be unlocked
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100646 let expected = [
Bram Moolenaara187c432020-09-16 21:08:28 +0200647 \ [['1000-000', 'ppppppp'],
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100648 \ ['0000-000', 'ppppppp'],
649 \ ['0000-000', 'ppppppp']],
650 \ [['1000-000', 'ppFppFp'],
651 \ ['0000-000', 'ppppppp'],
652 \ ['0000-000', 'ppppppp']],
653 \ [['1100-100', 'pFFpFFp'],
654 \ ['0000-000', 'ppppppp'],
655 \ ['0000-000', 'ppppppp']],
656 \ [['1110-110', 'FFFFFFp'],
657 \ ['0010-010', 'FppFppp'],
658 \ ['0000-000', 'ppppppp']],
659 \ [['1111-111', 'FFFFFFp'],
660 \ ['0011-011', 'FppFppp'],
661 \ ['0000-000', 'ppppppp']]
662 \ ]
663
664 for depth in range(5)
665 for u in range(3)
666 unlet! l
667 let l = [0, [1, [2, 3]], {4: 5, 6: {7: 8}}]
668 exe "lockvar " . depth . " l"
669 if u == 1
670 exe "unlockvar l"
671 elseif u == 2
672 exe "unlockvar " . depth . " l"
673 endif
674 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 +0200675 call assert_equal(expected[depth][u][0], ps, 'depth: ' .. depth)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100676 let ps = ''
677 try
678 unlet l[2]['6'][7]
679 let ps .= 'p'
680 catch
681 let ps .= 'F'
682 endtry
683 try
684 unlet l[2][6]
685 let ps .= 'p'
686 catch
687 let ps .= 'F'
688 endtry
689 try
690 unlet l[2]
691 let ps .= 'p'
692 catch
693 let ps .= 'F'
694 endtry
695 try
696 unlet l[1][1][0]
697 let ps .= 'p'
698 catch
699 let ps .= 'F'
700 endtry
701 try
702 unlet l[1][1]
703 let ps .= 'p'
704 catch
705 let ps .= 'F'
706 endtry
707 try
708 unlet l[1]
709 let ps .= 'p'
710 catch
711 let ps .= 'F'
712 endtry
713 try
714 unlet l
715 let ps .= 'p'
716 catch
717 let ps .= 'F'
718 endtry
719 call assert_equal(expected[depth][u][1], ps)
720 endfor
721 endfor
Yegappan Lakshmanan34fcb692021-05-25 20:14:00 +0200722 " Deleting a list range should fail if the range is locked
723 let l = [1, 2, 3, 4]
724 lockvar l[1:2]
725 call assert_fails('unlet l[1:2]', 'E741:')
726 unlet l
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100727endfunc
728
729" Locked variables and :unlet or list / dict functions
730
731" No :unlet after lock on dict:
732func Test_dict_lock_unlet()
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100733 let d = {'a': 99, 'b': 100}
734 lockvar 1 d
Bram Moolenaare2e40752020-09-04 21:18:46 +0200735 call assert_fails('unlet d.a', 'E741:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100736endfunc
737
738" unlet after lock on dict item
739func Test_dict_item_lock_unlet()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200740 let lines =<< trim END
741 VAR d = {'a': 99, 'b': 100}
742 lockvar d.a
743 unlet d.a
744 call assert_equal({'b': 100}, d)
745 END
Bram Moolenaar71b76852021-12-17 20:15:38 +0000746 call CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100747endfunc
748
749" filter() after lock on dict item
750func Test_dict_lock_filter()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200751 let lines =<< trim END
752 VAR d = {'a': 99, 'b': 100}
753 lockvar d.a
754 call filter(d, 'v:key != "a"')
755 call assert_equal({'b': 100}, d)
756 END
Bram Moolenaar71b76852021-12-17 20:15:38 +0000757 call CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100758endfunc
759
760" map() after lock on dict
761func Test_dict_lock_map()
Bram Moolenaarbd77aa92021-08-12 17:06:05 +0200762 let lines =<< trim END
763 VAR d = {'a': 99, 'b': 100}
764 lockvar 1 d
765 call map(d, 'v:val + 200')
766 call assert_equal({'a': 299, 'b': 300}, d)
767 END
768 " This won't work in a :def function
769 call CheckTransLegacySuccess(lines)
770 call CheckTransVim9Success(lines)
Bram Moolenaar71b76852021-12-17 20:15:38 +0000771
772 " For a :def function use a global dict.
773 let lines =<< trim END
774 let g:thedict = {'a': 77, 'b': 88}
775 lockvar 1 g:thedict
776 def Delkey()
777 unlet g:thedict.a
778 enddef
779 call Delkey()
780 END
781 call CheckScriptFailure(lines, 'E741:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100782endfunc
783
784" No extend() after lock on dict item
785func Test_dict_lock_extend()
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100786 let d = {'a': 99, 'b': 100}
787 lockvar d.a
Bram Moolenaare2e40752020-09-04 21:18:46 +0200788 call assert_fails("call extend(d, {'a' : 123})", 'E741:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100789 call assert_equal({'a': 99, 'b': 100}, d)
790endfunc
791
Bram Moolenaarf7b398c2020-04-23 15:46:35 +0200792" Cannot use += with a locked dict
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200793func Test_dict_lock_operator()
Bram Moolenaarea04a6e2020-04-23 13:38:02 +0200794 let d = {}
795 lockvar d
796 call assert_fails("let d += {'k' : 10}", 'E741:')
797 unlockvar d
798endfunc
799
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100800" No remove() of write-protected scope-level variable
Bram Moolenaar1e115362019-01-09 23:01:02 +0100801func Tfunc1(this_is_a_long_parameter_name)
Bram Moolenaare2e40752020-09-04 21:18:46 +0200802 call assert_fails("call remove(a:, 'this_is_a_long_parameter_name')", 'E742:')
Bram Moolenaar1e115362019-01-09 23:01:02 +0100803endfunc
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100804func Test_dict_scope_var_remove()
Bram Moolenaar1e115362019-01-09 23:01:02 +0100805 call Tfunc1('testval')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100806endfunc
807
808" No extend() of write-protected scope-level variable
Bram Moolenaar31b81602019-02-10 22:14:27 +0100809func Test_dict_scope_var_extend()
Bram Moolenaare2e40752020-09-04 21:18:46 +0200810 call assert_fails("call extend(a:, {'this_is_a_long_parameter_name': 1234})", 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +0100811endfunc
812
Bram Moolenaar1e115362019-01-09 23:01:02 +0100813func Tfunc2(this_is_a_long_parameter_name)
Bram Moolenaare2e40752020-09-04 21:18:46 +0200814 call assert_fails("call extend(a:, {'this_is_a_long_parameter_name': 1234})", 'E742:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100815endfunc
Bram Moolenaar31b81602019-02-10 22:14:27 +0100816func Test_dict_scope_var_extend_overwrite()
Bram Moolenaar1e115362019-01-09 23:01:02 +0100817 call Tfunc2('testval')
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100818endfunc
819
820" No :unlet of variable in locked scope
821func Test_lock_var_unlet()
822 let b:testvar = 123
823 lockvar 1 b:
824 call assert_fails('unlet b:testvar', 'E741:')
825 unlockvar 1 b:
826 unlet! b:testvar
827endfunc
828
829" No :let += of locked list variable
830func Test_let_lock_list()
831 let l = ['a', 'b', 3]
832 lockvar 1 l
833 call assert_fails("let l += ['x']", 'E741:')
834 call assert_equal(['a', 'b', 3], l)
835
836 unlet l
837 let l = [1, 2, 3, 4]
838 lockvar! l
839 call assert_equal([1, 2, 3, 4], l)
840 unlockvar l[1]
841 call assert_fails('unlet l[0:1]', 'E741:')
842 call assert_equal([1, 2, 3, 4], l)
843 call assert_fails('unlet l[1:2]', 'E741:')
844 call assert_equal([1, 2, 3, 4], l)
845 unlockvar l[1]
846 call assert_fails('let l[0:1] = [0, 1]', 'E741:')
847 call assert_equal([1, 2, 3, 4], l)
848 call assert_fails('let l[1:2] = [0, 1]', 'E741:')
849 call assert_equal([1, 2, 3, 4], l)
850 unlet l
851endfunc
852
Bram Moolenaar8dfcce32020-03-18 19:32:26 +0100853" Locking part of the list
854func Test_let_lock_list_items()
855 let l = [1, 2, 3, 4]
856 lockvar l[2:]
857 call assert_equal(0, islocked('l[0]'))
858 call assert_equal(1, islocked('l[2]'))
859 call assert_equal(1, islocked('l[3]'))
860 call assert_fails('let l[2] = 10', 'E741:')
861 call assert_fails('let l[3] = 20', 'E741:')
862 unlet l
863endfunc
864
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100865" lockvar/islocked() triggering script autoloading
866func Test_lockvar_script_autoload()
867 let old_rtp = &rtp
868 set rtp+=./sautest
869 lockvar g:footest#x
870 unlockvar g:footest#x
Bram Moolenaarf9f24ce2019-08-31 21:17:39 +0200871 call assert_equal(-1, 'g:footest#x'->islocked())
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100872 call assert_equal(0, exists('g:footest#x'))
873 call assert_equal(1, g:footest#x)
874 let &rtp = old_rtp
875endfunc
876
877" a:000 function argument test
878func s:arg_list_test(...)
879 call assert_fails('let a:000 = [1, 2]', 'E46:')
880 call assert_fails('let a:000[0] = 9', 'E742:')
881 call assert_fails('let a:000[2] = [9, 10]', 'E742:')
882 call assert_fails('let a:000[3] = {9 : 10}', 'E742:')
883
884 " now the tests that should pass
885 let a:000[2][1] = 9
886 call extend(a:000[2], [5, 6])
887 let a:000[3][5] = 8
888 let a:000[3]['a'] = 12
889 call assert_equal([1, 2, [3, 9, 5, 6], {'a': 12, '5': 8}], a:000)
890endfunc
891
892func Test_func_arg_list()
893 call s:arg_list_test(1, 2, [3, 4], {5: 6})
894endfunc
895
896" Tests for reverse(), sort(), uniq()
897func Test_reverse_sort_uniq()
Bram Moolenaaref982572021-08-12 19:27:57 +0200898 let lines =<< trim END
899 VAR l = ['-0', 'A11', 2, 2, 'xaaa', 4, 'foo', 'foo6', 'foo', [0, 1, 2], 'x8', [0, 1, 2], 1.5]
900 call assert_equal(['-0', 'A11', 2, 'xaaa', 4, 'foo', 'foo6', 'foo', [0, 1, 2], 'x8', [0, 1, 2], 1.5], uniq(copy(l)))
901 call assert_equal([1.5, [0, 1, 2], 'x8', [0, 1, 2], 'foo', 'foo6', 'foo', 4, 'xaaa', 2, 2, 'A11', '-0'], reverse(l))
902 call assert_equal([1.5, [0, 1, 2], 'x8', [0, 1, 2], 'foo', 'foo6', 'foo', 4, 'xaaa', 2, 2, 'A11', '-0'], reverse(reverse(l)))
903 if has('float')
904 call assert_equal(['-0', 'A11', 'foo', 'foo', 'foo6', 'x8', 'xaaa', 1.5, 2, 2, 4, [0, 1, 2], [0, 1, 2]], sort(l))
905 call assert_equal([[0, 1, 2], [0, 1, 2], 4, 2, 2, 1.5, 'xaaa', 'x8', 'foo6', 'foo', 'foo', 'A11', '-0'], reverse(sort(l)))
906 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))))
907 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 +0100908
Bram Moolenaaref982572021-08-12 19:27:57 +0200909 LET l = [7, 9, 'one', 18, 12, 22, 'two', 10.0e-16, -1, 'three', 0xff, 0.22, 'four']
910 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 +0100911
Bram Moolenaaref982572021-08-12 19:27:57 +0200912 LET l = [7, 9, 18, 12, 22, 10.0e-16, -1, 0xff, 0, -0, 0.22, 'bar', 'BAR', 'Bar', 'Foo', 'FOO', 'foo', 'FOOBAR', {}, []]
913 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))
914 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'))
915 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)))
916 endif
917 END
918 call CheckLegacyAndVim9Success(lines)
Bram Moolenaar0d17f0d2019-01-22 22:20:38 +0100919
Bram Moolenaarbf821bc2019-01-23 21:15:02 +0100920 call assert_fails('call reverse("")', 'E899:')
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200921 call assert_fails('call uniq([1, 2], {x, y -> []})', 'E745:')
Bram Moolenaar08f41572020-04-20 16:50:00 +0200922 call assert_fails("call sort([1, 2], function('min'), 1)", "E715:")
923 call assert_fails("call sort([1, 2], function('invalid_func'))", "E700:")
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200924 call assert_fails("call sort([1, 2], function('min'))", "E118:")
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100925endfunc
926
Bram Moolenaar85629982020-06-01 18:39:20 +0200927" reduce a list or a blob
928func Test_reduce()
Bram Moolenaaref982572021-08-12 19:27:57 +0200929 let lines =<< trim END
930 call assert_equal(1, reduce([], LSTART acc, val LMIDDLE acc + val LEND, 1))
931 call assert_equal(10, reduce([1, 3, 5], LSTART acc, val LMIDDLE acc + val LEND, 1))
932 call assert_equal(2 * (2 * ((2 * 1) + 2) + 3) + 4, reduce([2, 3, 4], LSTART acc, val LMIDDLE 2 * acc + val LEND, 1))
933 call assert_equal('a x y z', ['x', 'y', 'z']->reduce(LSTART acc, val LMIDDLE acc .. ' ' .. val LEND, 'a'))
934 call assert_equal([0, 1, 2, 3], reduce([1, 2, 3], function('add'), [0]))
Bram Moolenaar85629982020-06-01 18:39:20 +0200935
Bram Moolenaaref982572021-08-12 19:27:57 +0200936 VAR l = ['x', 'y', 'z']
937 call assert_equal(42, reduce(l, function('get'), {'x': {'y': {'z': 42 } } }))
938 call assert_equal(['x', 'y', 'z'], l)
Bram Moolenaar85629982020-06-01 18:39:20 +0200939
Bram Moolenaaref982572021-08-12 19:27:57 +0200940 call assert_equal(1, reduce([1], LSTART acc, val LMIDDLE acc + val LEND))
941 call assert_equal('x y z', reduce(['x', 'y', 'z'], LSTART acc, val LMIDDLE acc .. ' ' .. val LEND))
942 call assert_equal(120, range(1, 5)->reduce(LSTART acc, val LMIDDLE acc * val LEND))
943
944 call assert_equal(1, reduce(0z, LSTART acc, val LMIDDLE acc + val LEND, 1))
945 call assert_equal(1 + 0xaf + 0xbf + 0xcf, reduce(0zAFBFCF, LSTART acc, val LMIDDLE acc + val LEND, 1))
946 call assert_equal(2 * (2 * 1 + 0xaf) + 0xbf, 0zAFBF->reduce(LSTART acc, val LMIDDLE 2 * acc + val LEND, 1))
947
948 call assert_equal(0xff, reduce(0zff, LSTART acc, val LMIDDLE acc + val LEND))
949 call assert_equal(2 * (2 * 0xaf + 0xbf) + 0xcf, reduce(0zAFBFCF, LSTART acc, val LMIDDLE 2 * acc + val LEND))
950 END
951 call CheckLegacyAndVim9Success(lines)
952
953 call assert_equal({'x': 1, 'y': 1, 'z': 1 }, ['x', 'y', 'z']->reduce({ acc, val -> extend(acc, { val: 1 }) }, {}))
954 vim9 assert_equal({'x': 1, 'y': 1, 'z': 1 }, ['x', 'y', 'z']->reduce((acc, val) => extend(acc, {[val]: 1 }), {}))
955
Bram Moolenaar85629982020-06-01 18:39:20 +0200956 call assert_fails("call reduce([], { acc, val -> acc + val })", 'E998: Reduce of an empty List with no initial value')
Bram Moolenaar85629982020-06-01 18:39:20 +0200957 call assert_fails("call reduce(0z, { acc, val -> acc + val })", 'E998: Reduce of an empty Blob with no initial value')
958
959 call assert_fails("call reduce({}, { acc, val -> acc + val }, 1)", 'E897:')
960 call assert_fails("call reduce(0, { acc, val -> acc + val }, 1)", 'E897:')
961 call assert_fails("call reduce('', { acc, val -> acc + val }, 1)", 'E897:')
Bram Moolenaarafe8cf62020-10-05 20:07:18 +0200962 call assert_fails("call reduce([1, 2], 'Xdoes_not_exist')", 'E117:')
963 call assert_fails("echo reduce(0z01, { acc, val -> 2 * acc + val }, '')", 'E39:')
Bram Moolenaarca275a02020-06-24 22:07:46 +0200964
965 let g:lut = [1, 2, 3, 4]
966 func EvilRemove()
967 call remove(g:lut, 1)
968 return 1
969 endfunc
970 call assert_fails("call reduce(g:lut, { acc, val -> EvilRemove() }, 1)", 'E742:')
971 unlet g:lut
972 delfunc EvilRemove
Bram Moolenaarfda20c42020-06-29 20:09:36 +0200973
974 call assert_equal(42, reduce(test_null_list(), function('add'), 42))
975 call assert_equal(42, reduce(test_null_blob(), function('add'), 42))
Bram Moolenaar0d90e722020-11-03 18:20:19 +0100976
977 " should not crash
978 call assert_fails('echo reduce([1], test_null_function())', 'E1132:')
Dominique Pellefe8ebdb2021-05-13 14:55:55 +0200979 call assert_fails('echo reduce([1], test_null_partial())', 'E1132:')
Bram Moolenaar85629982020-06-01 18:39:20 +0200980endfunc
981
Bram Moolenaar0e05de42020-03-25 22:23:46 +0100982" splitting a string to a List using split()
Bram Moolenaarfb094e12017-11-05 20:59:28 +0100983func Test_str_split()
Bram Moolenaaref982572021-08-12 19:27:57 +0200984 let lines =<< trim END
985 call assert_equal(['aa', 'bb'], split(' aa bb '))
986 call assert_equal(['aa', 'bb'], split(' aa bb ', '\W\+', 0))
987 call assert_equal(['', 'aa', 'bb', ''], split(' aa bb ', '\W\+', 1))
988 call assert_equal(['', '', 'aa', '', 'bb', '', ''], split(' aa bb ', '\W', 1))
989 call assert_equal(['aa', '', 'bb'], split(':aa::bb:', ':', 0))
990 call assert_equal(['', 'aa', '', 'bb', ''], split(':aa::bb:', ':', 1))
991 call assert_equal(['aa', '', 'bb', 'cc', ''], split('aa,,bb, cc,', ',\s*', 1))
992 call assert_equal(['a', 'b', 'c'], split('abc', '\zs'))
993 call assert_equal(['', 'a', '', 'b', '', 'c', ''], split('abc', '\zs', 1))
994 call assert_equal(['abc'], split('abc', '\\%('))
995 END
996 call CheckLegacyAndVim9Success(lines)
997
Bram Moolenaar0e05de42020-03-25 22:23:46 +0100998 call assert_fails("call split('abc', [])", 'E730:')
999 call assert_fails("call split('abc', 'b', [])", 'E745:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001000endfunc
1001
1002" compare recursively linked list and dict
1003func Test_listdict_compare()
Bram Moolenaaref982572021-08-12 19:27:57 +02001004 let lines =<< trim END
1005 VAR l = [1, 2, 3, '4']
1006 VAR d = {'1': 1, '2': l, '3': 3}
1007 LET l[1] = d
1008 call assert_true(l == l)
1009 call assert_true(d == d)
1010 call assert_false(l != deepcopy(l))
1011 call assert_false(d != deepcopy(d))
1012 END
1013 call CheckLegacyAndVim9Success(lines)
Bram Moolenaar8b633132020-03-20 18:20:51 +01001014
1015 " comparison errors
1016 call assert_fails('echo [1, 2] =~ {}', 'E691:')
1017 call assert_fails('echo [1, 2] =~ [1, 2]', 'E692:')
1018 call assert_fails('echo {} =~ 5', 'E735:')
1019 call assert_fails('echo {} =~ {}', 'E736:')
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001020endfunc
1021
1022 " compare complex recursively linked list and dict
1023func Test_listdict_compare_complex()
Bram Moolenaaref982572021-08-12 19:27:57 +02001024 let lines =<< trim END
1025 VAR l = []
1026 call add(l, l)
1027 VAR dict4 = {"l": l}
1028 call add(dict4.l, dict4)
1029 VAR lcopy = deepcopy(l)
1030 VAR dict4copy = deepcopy(dict4)
1031 call assert_true(l == lcopy)
1032 call assert_true(dict4 == dict4copy)
1033 END
1034 call CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001035endfunc
1036
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001037" Test for extending lists and dictionaries
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001038func Test_listdict_extend()
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001039 " Test extend() with lists
1040
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001041 " Pass the same List to extend()
Bram Moolenaaref982572021-08-12 19:27:57 +02001042 let lines =<< trim END
1043 VAR l = [1, 2, 3]
1044 call assert_equal([1, 2, 3, 1, 2, 3], extend(l, l))
1045 call assert_equal([1, 2, 3, 1, 2, 3], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001046
Bram Moolenaaref982572021-08-12 19:27:57 +02001047 LET l = [1, 2, 3]
1048 call assert_equal([1, 2, 3, 4, 5, 6], extend(l, [4, 5, 6]))
1049 call assert_equal([1, 2, 3, 4, 5, 6], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001050
Bram Moolenaaref982572021-08-12 19:27:57 +02001051 LET l = [1, 2, 3]
1052 call extend(l, [4, 5, 6], 0)
1053 call assert_equal([4, 5, 6, 1, 2, 3], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001054
Bram Moolenaaref982572021-08-12 19:27:57 +02001055 LET l = [1, 2, 3]
1056 call extend(l, [4, 5, 6], 1)
1057 call assert_equal([1, 4, 5, 6, 2, 3], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001058
Bram Moolenaaref982572021-08-12 19:27:57 +02001059 LET l = [1, 2, 3]
1060 call extend(l, [4, 5, 6], 3)
1061 call assert_equal([1, 2, 3, 4, 5, 6], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001062
Bram Moolenaaref982572021-08-12 19:27:57 +02001063 LET l = [1, 2, 3]
1064 call extend(l, [4, 5, 6], -1)
1065 call assert_equal([1, 2, 4, 5, 6, 3], l)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001066
Bram Moolenaaref982572021-08-12 19:27:57 +02001067 LET l = [1, 2, 3]
1068 call extend(l, [4, 5, 6], -3)
1069 call assert_equal([4, 5, 6, 1, 2, 3], l)
1070 END
1071 call CheckLegacyAndVim9Success(lines)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001072
1073 let l = [1, 2, 3]
1074 call assert_fails("call extend(l, [4, 5, 6], 4)", 'E684:')
1075 call assert_fails("call extend(l, [4, 5, 6], -4)", 'E684:')
Bram Moolenaar5feabe02020-01-30 18:24:53 +01001076 if has('float')
1077 call assert_fails("call extend(l, [4, 5, 6], 1.2)", 'E805:')
1078 endif
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001079
1080 " Test extend() with dictionaries.
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001081
1082 " Pass the same Dict to extend()
Bram Moolenaaref982572021-08-12 19:27:57 +02001083 let lines =<< trim END
1084 VAR d = {'a': {'b': 'B'}, 'x': 9}
1085 call extend(d, d)
1086 call assert_equal({'a': {'b': 'B'}, 'x': 9}, d)
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001087
Bram Moolenaaref982572021-08-12 19:27:57 +02001088 LET d = {'a': 'A', 'b': 9}
1089 call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, extend(d, {'b': 0, 'c': 'C'}))
1090 call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, d)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001091
Bram Moolenaaref982572021-08-12 19:27:57 +02001092 LET d = {'a': 'A', 'b': 9}
1093 call extend(d, {'a': 'A', 'b': 0, 'c': 'C'}, "force")
1094 call assert_equal({'a': 'A', 'b': 0, 'c': 'C'}, d)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001095
Bram Moolenaaref982572021-08-12 19:27:57 +02001096 LET d = {'a': 'A', 'b': 9}
1097 call extend(d, {'b': 0, 'c': 'C'}, "keep")
1098 call assert_equal({'a': 'A', 'b': 9, 'c': 'C'}, d)
1099 END
1100 call CheckLegacyAndVim9Success(lines)
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001101
1102 let d = {'a': 'A', 'b': 'B'}
1103 call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 'error')", 'E737:')
1104 call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 'xxx')", 'E475:')
Bram Moolenaar5feabe02020-01-30 18:24:53 +01001105 if has('float')
Bram Moolenaar3cfa5b12021-06-06 14:14:39 +02001106 call assert_fails("call extend(d, {'b': 0, 'c':'C'}, 1.2)", 'E475:')
Bram Moolenaar5feabe02020-01-30 18:24:53 +01001107 endif
Bram Moolenaar58d63a02019-02-25 05:56:31 +01001108 call assert_equal({'a': 'A', 'b': 'B'}, d)
1109
1110 call assert_fails("call extend([1, 2], 1)", 'E712:')
1111 call assert_fails("call extend([1, 2], {})", 'E712:')
Bram Moolenaar08f41572020-04-20 16:50:00 +02001112
1113 " Extend g: dictionary with an invalid variable name
1114 call assert_fails("call extend(g:, {'-!' : 10})", 'E461:')
Bram Moolenaardcae51f2021-04-08 20:10:10 +02001115
1116 " Extend a list with itself.
Bram Moolenaaref982572021-08-12 19:27:57 +02001117 let lines =<< trim END
1118 VAR l = [1, 5, 7]
1119 call extend(l, l, 0)
1120 call assert_equal([1, 5, 7, 1, 5, 7], l)
1121 LET l = [1, 5, 7]
1122 call extend(l, l, 1)
1123 call assert_equal([1, 1, 5, 7, 5, 7], l)
1124 LET l = [1, 5, 7]
1125 call extend(l, l, 2)
1126 call assert_equal([1, 5, 1, 5, 7, 7], l)
1127 LET l = [1, 5, 7]
1128 call extend(l, l, 3)
1129 call assert_equal([1, 5, 7, 1, 5, 7], l)
1130 END
1131 call CheckLegacyAndVim9Success(lines)
Bram Moolenaarfb094e12017-11-05 20:59:28 +01001132endfunc
Bram Moolenaar31b81602019-02-10 22:14:27 +01001133
Bram Moolenaarb0e6b512021-01-12 20:23:40 +01001134func Test_listdict_extendnew()
1135 " Test extendnew() with lists
1136 let l = [1, 2, 3]
1137 call assert_equal([1, 2, 3, 4, 5], extendnew(l, [4, 5]))
1138 call assert_equal([1, 2, 3], l)
1139
1140 " Test extend() with dictionaries.
1141 let d = {'a': {'b': 'B'}}
1142 call assert_equal({'a': {'b': 'B'}, 'c': 'cc'}, extendnew(d, {'c': 'cc'}))
1143 call assert_equal({'a': {'b': 'B'}}, d)
1144endfunc
1145
Bram Moolenaar31b81602019-02-10 22:14:27 +01001146func s:check_scope_dict(x, fixed)
1147 func s:gen_cmd(cmd, x)
1148 return substitute(a:cmd, '\<x\ze:', a:x, 'g')
1149 endfunc
1150
1151 let cmd = s:gen_cmd('let x:foo = 1', a:x)
1152 if a:fixed
Bram Moolenaare2e40752020-09-04 21:18:46 +02001153 call assert_fails(cmd, 'E461:')
Bram Moolenaar31b81602019-02-10 22:14:27 +01001154 else
1155 exe cmd
1156 exe s:gen_cmd('call assert_equal(1, x:foo)', a:x)
1157 endif
1158
1159 let cmd = s:gen_cmd('let x:["bar"] = 2', a:x)
1160 if a:fixed
Bram Moolenaare2e40752020-09-04 21:18:46 +02001161 call assert_fails(cmd, 'E461:')
Bram Moolenaar31b81602019-02-10 22:14:27 +01001162 else
1163 exe cmd
1164 exe s:gen_cmd('call assert_equal(2, x:bar)', a:x)
1165 endif
1166
1167 let cmd = s:gen_cmd('call extend(x:, {"baz": 3})', a:x)
1168 if a:fixed
Bram Moolenaare2e40752020-09-04 21:18:46 +02001169 call assert_fails(cmd, 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +01001170 else
1171 exe cmd
1172 exe s:gen_cmd('call assert_equal(3, x:baz)', a:x)
1173 endif
1174
1175 if a:fixed
1176 if a:x ==# 'a'
Bram Moolenaare2e40752020-09-04 21:18:46 +02001177 call assert_fails('unlet a:x', 'E795:')
1178 call assert_fails('call remove(a:, "x")', 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +01001179 elseif a:x ==# 'v'
Bram Moolenaare2e40752020-09-04 21:18:46 +02001180 call assert_fails('unlet v:count', 'E795:')
1181 call assert_fails('call remove(v:, "count")', 'E742:')
Bram Moolenaar31b81602019-02-10 22:14:27 +01001182 endif
1183 else
1184 exe s:gen_cmd('unlet x:foo', a:x)
1185 exe s:gen_cmd('unlet x:bar', a:x)
1186 exe s:gen_cmd('call remove(x:, "baz")', a:x)
1187 endif
1188
1189 delfunc s:gen_cmd
1190endfunc
1191
1192func Test_scope_dict()
1193 " Test for g:
1194 call s:check_scope_dict('g', v:false)
1195
1196 " Test for s:
1197 call s:check_scope_dict('s', v:false)
1198
1199 " Test for l:
1200 call s:check_scope_dict('l', v:false)
1201
1202 " Test for a:
1203 call s:check_scope_dict('a', v:true)
1204
1205 " Test for b:
1206 call s:check_scope_dict('b', v:false)
1207
1208 " Test for w:
1209 call s:check_scope_dict('w', v:false)
1210
1211 " Test for t:
1212 call s:check_scope_dict('t', v:false)
1213
1214 " Test for v:
1215 call s:check_scope_dict('v', v:true)
1216endfunc
Bram Moolenaar8dfcce32020-03-18 19:32:26 +01001217
1218" Test for deep nesting of lists (> 100)
1219func Test_deep_nested_list()
1220 let deep_list = []
1221 let l = deep_list
1222 for i in range(102)
1223 let newlist = []
1224 call add(l, newlist)
1225 let l = newlist
1226 endfor
1227 call add(l, 102)
1228
1229 call assert_fails('let m = deepcopy(deep_list)', 'E698:')
1230 call assert_fails('lockvar 110 deep_list', 'E743:')
1231 call assert_fails('unlockvar 110 deep_list', 'E743:')
1232 call assert_fails('let x = execute("echo deep_list")', 'E724:')
1233 call test_garbagecollect_now()
1234 unlet deep_list
1235endfunc
1236
1237" Test for deep nesting of dicts (> 100)
1238func Test_deep_nested_dict()
1239 let deep_dict = {}
1240 let d = deep_dict
1241 for i in range(102)
1242 let newdict = {}
1243 let d.k = newdict
1244 let d = newdict
1245 endfor
1246 let d.k = 'v'
1247
1248 call assert_fails('let m = deepcopy(deep_dict)', 'E698:')
1249 call assert_fails('lockvar 110 deep_dict', 'E743:')
1250 call assert_fails('unlockvar 110 deep_dict', 'E743:')
1251 call assert_fails('let x = execute("echo deep_dict")', 'E724:')
1252 call test_garbagecollect_now()
1253 unlet deep_dict
1254endfunc
1255
Bram Moolenaar8b633132020-03-20 18:20:51 +01001256" List and dict indexing tests
1257func Test_listdict_index()
1258 call assert_fails('echo function("min")[0]', 'E695:')
1259 call assert_fails('echo v:true[0]', 'E909:')
1260 let d = {'k' : 10}
1261 call assert_fails('echo d.', 'E15:')
1262 call assert_fails('echo d[1:2]', 'E719:')
1263 call assert_fails("let v = [4, 6][{-> 1}]", 'E729:')
1264 call assert_fails("let v = range(5)[2:[]]", 'E730:')
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001265 call assert_fails("let v = range(5)[2:{-> 2}(]", ['E15:', 'E116:'])
Bram Moolenaar8b633132020-03-20 18:20:51 +01001266 call assert_fails("let v = range(5)[2:3", 'E111:')
Bram Moolenaar08f41572020-04-20 16:50:00 +02001267 call assert_fails("let l = insert([1,2,3], 4, 10)", 'E684:')
1268 call assert_fails("let l = insert([1,2,3], 4, -10)", 'E684:')
1269 call assert_fails("let l = insert([1,2,3], 4, [])", 'E745:')
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001270 let l = [1, 2, 3]
1271 call assert_fails("let l[i] = 3", 'E121:')
Bram Moolenaar3cfa5b12021-06-06 14:14:39 +02001272 call assert_fails("let l[1.1] = 4", 'E805:')
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001273 call assert_fails("let l[:i] = [4, 5]", 'E121:')
Bram Moolenaar3cfa5b12021-06-06 14:14:39 +02001274 call assert_fails("let l[:3.2] = [4, 5]", 'E805:')
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001275 let t = test_unknown()
1276 call assert_fails("echo t[0]", 'E685:')
Bram Moolenaar08f41572020-04-20 16:50:00 +02001277endfunc
1278
1279" Test for a null list
1280func Test_null_list()
Bram Moolenaaref982572021-08-12 19:27:57 +02001281 let lines =<< trim END
1282 VAR l = test_null_list()
1283 call assert_equal('', join(test_null_list()))
1284 call assert_equal('', join(l))
1285 call assert_equal(0, len(l))
1286 call assert_equal(1, empty(l))
1287 call assert_equal([], split(test_null_string()))
1288 call assert_equal([], l[ : 2])
1289 call assert_true([] == l)
1290 call assert_equal('[]', string(l))
1291 call assert_equal([], sort(test_null_list()))
1292 call assert_equal([], sort(l))
1293 call assert_equal([], uniq(test_null_list()))
1294 call assert_equal([], uniq(l))
1295 VAR k = [] + l
1296 call assert_equal([], k)
1297 LET k = l + []
1298 call assert_equal([], k)
1299 call assert_equal(0, len(copy(l)))
1300 call assert_equal(0, count(l, 5))
1301 call assert_equal([], deepcopy(l))
1302 call assert_equal(5, get(l, 2, 5))
1303 call assert_equal(-1, index(l, 2, 5))
1304 call assert_equal(0, min(l))
1305 call assert_equal(0, max(l))
1306 call assert_equal(0, remove(test_null_list(), 0, 2))
1307 call assert_equal([], repeat(l, 2))
1308 call assert_equal([], reverse(test_null_list()))
1309 call assert_equal([], reverse(l))
1310 call assert_equal([], sort(test_null_list()))
1311 call assert_equal([], sort(l))
1312 call assert_equal('[]', string(l))
1313 END
1314 call CheckLegacyAndVim9Success(lines)
1315
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001316 let l = test_null_list()
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001317 call assert_equal([], extend(l, l, 0))
Bram Moolenaaref982572021-08-12 19:27:57 +02001318 call assert_equal(0, insert(test_null_list(), 2, -1))
1319 call assert_fails('let s = join([1, 2], [])', 'E730:')
1320 call assert_fails('call remove(l, 0, 2)', 'E684:')
1321 call assert_fails('call insert(l, 2, -1)', 'E684:')
1322 call assert_fails('call extend(test_null_list(), test_null_list())', 'E1134:')
1323
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001324 lockvar l
1325 call assert_equal(1, islocked('l'))
1326 unlockvar l
Bram Moolenaar08f41572020-04-20 16:50:00 +02001327endfunc
1328
1329" Test for a null dict
1330func Test_null_dict()
Bram Moolenaaref982572021-08-12 19:27:57 +02001331 let lines =<< trim END
1332 call assert_equal(test_null_dict(), test_null_dict())
1333 VAR d = test_null_dict()
1334 call assert_equal({}, d)
1335 call assert_equal(0, len(d))
1336 call assert_equal(1, empty(d))
1337 call assert_equal([], items(test_null_dict()))
1338 call assert_equal([], items(d))
1339 call assert_equal([], keys(test_null_dict()))
1340 call assert_equal([], keys(d))
1341 call assert_equal([], values(test_null_dict()))
1342 call assert_equal([], values(d))
1343 call assert_false(has_key(d, 'k'))
1344 call assert_equal('{}', string(d))
1345 call assert_equal({}, {})
1346 call assert_equal(0, len(copy(d)))
1347 call assert_equal(0, count(d, 'k'))
1348 call assert_equal({}, deepcopy(d))
1349 call assert_equal(20, get(d, 'k', 20))
1350 call assert_equal(0, min(d))
1351 call assert_equal(0, max(d))
1352 call assert_equal(0, remove(test_null_dict(), 'k'))
1353 call assert_equal('{}', string(d))
1354 END
1355 call CheckLegacyAndVim9Success(lines)
1356
Bram Moolenaarea04a6e2020-04-23 13:38:02 +02001357 let d = test_null_dict()
Bram Moolenaar64ffa9b2020-11-04 12:23:06 +01001358 call assert_equal({}, extend(d, d, 'keep'))
Bram Moolenaaref982572021-08-12 19:27:57 +02001359 call assert_fails("call remove(d, 'k')", 'E716:')
1360 call assert_fails('let x = d[10]', 'E716:')
1361 call assert_fails('call extend(test_null_dict(), test_null_dict())', 'E1133:')
Bram Moolenaar92b83cc2020-04-25 15:24:44 +02001362 lockvar d
1363 call assert_equal(1, islocked('d'))
1364 unlockvar d
Bram Moolenaar8b633132020-03-20 18:20:51 +01001365endfunc
1366
Bram Moolenaar8dfcce32020-03-18 19:32:26 +01001367" vim: shiftwidth=2 sts=2 expandtab